tensorflow版本Faster R-CNN特征图可视化
参考文献:
https://blog.****.net/zd_nupt/article/details/77508083
endernewton 的 Tensorflow 版 Faster RCNN 代码:
https://github.com/endernewton/tf-faster-rcnn
主要代码:
with tf.Session() as sess:#打开会话
sess.run(init)#初始化
img1 = mnist.train.images[1]#输入图片
img1.shape = [1, 784]#转换接口格式
result = sess.run(h_conv1, feed_dict={xs : img1})#提取tensor特征
for i in range(32):#画出每一张特征图
show_img = result[:, :, :, i]
# print type(show_img)
show_img.shape = [28, 28]
plt.subplot(4, 8, i + 1)#打开一个4x8的画布。
plt.imshow(show_img, cmap='gray')
plt.axis('off')
plt.show()
效果:
单张显示:
集体显示: