Tensorflow2.0 查看网络中每层的名称、权重及特征图绘制

项目介绍

在网络训练过程中,我们经常需要查看某层权重的变化过程,这其实只需要简单的 API 就能实现。

为了方便演示,我们使用迁移学习到的 MobileNet V2 网络。

实现过程

1、构建网络

我们将冻结迁移到的 MobileNet V2 网络,然后将它最后的分类层换成我们自己定义的分类层即可。

mobile = tf.keras.applications.MobileNetV2(include_top=False, weights

版权声明:本文为qq_36758914原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
THE END
< <上一篇
下一篇>>