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