tensorboard可视化探索

调试网络模型时,为了了解每一层的特征提取情况,可采用tf.summary.image查看每一输出层的输出特征

查看某一网络层输出的关键代码:

    writer = tf.summary.FileWriter('logs') #将输出写入到logs文件夹中
    summary_op = tf.summary.image("image", tmp, max_outputs=1)#"image"为该图像的取名,
    #可以自己随意变换,一般设为网络层的tensor_name,tmp表示你要输出的内容
    summary = sess.run(summary_op)
    writer.add_summary(summary)

以下是查看模型的inputs层与自己的输入图像是否一致

              用imshow查看的图                                                          用tf.summary.image查看的图

ps:第一次知道inputs层的图像经过了横向的翻转

发布了46 篇原创文章 · 获赞 17 · 访问量 5万+

猜你喜欢

转载自blog.csdn.net/Jum_Summer/article/details/100584739
今日推荐