文章目录 项目介绍 实现过程 1、构建网络 2、查看每层名称 3、查看指定层的权值 4、特征图绘制 项目介绍 在网络训练过程中,我们经常需要查看某层权重的变化过程,这其实只需要简单的 API 就能实现。 为了方便演示,我们使用迁移学习到的 MobileNet V2 网络。 实现过程 1、构建网络 我们将冻结迁移到的 MobileNet V2 网络,然后将它最后的分类层换成我们自己定义的分类层即可。 mobile = tf.keras.applications.MobileNetV2(include_top=False, weights='imagenet