Tensorflow2.0 查看网络中每层的名称、权重及特征图绘制
项目介绍
在网络训练过程中,我们经常需要查看某层权重的变化过程,这其实只需要简单的 API 就能实现。
为了方便演示,我们使用迁移学习到的 MobileNet V2 网络。
实现过程
1、构建网络
我们将冻结迁移到的 MobileNet V2 网络,然后将它最后的分类层换成我们自己定义的分类层即可。
mobile = tf.keras.applications.MobileNetV2(include_top=False, weights
版权声明:本文为qq_36758914原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。