调试网络模型时,为了了解每一层的特征提取情况,可采用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层的图像经过了横向的翻转