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()

效果:

单张显示:

tensorflow版本Faster R-CNN特征图可视化

集体显示:

tensorflow版本Faster R-CNN特征图可视化