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

文章目录

  • 项目介绍
  • 实现过程
    • 1、构建网络
    • 2、查看每层名称
    • 3、查看指定层的权值
    • 4、特征图绘制

项目介绍

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

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

实现过程

1、构建网络

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

mobile = tf.keras.applications.MobileNetV2(include_top=

你可能感兴趣的:(Tensorflow,2.0,深度学习,PHM项目实战--建模篇,深度学习,python,tensorflow)