faster r-cnn批量测试图片及保存

1.批量测试图片

test_image_path = r'E:\DeepLearning\Faster-RCNN-TensorFlow-Python3.5-master\data\demo\im_names'   #要测试的图片放入im_names文件夹,并将改文件夹放入demo中 
im_names = os.listdir(test_image_path)
for im_name in im_names:
print('~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~')
print('Demo for data/demo/{}'.format(im_name))
demo(sess, net, im_name)
plt.savefig('C:\\Users\\wy\\Desktop\\new\\out' + im_name) #测试结果保存路径及名称(在new文件中再建一个目录,最终保存结果为out000001形式)
#plt.show()
2.在一张图片上显示所有的类
https://blog.csdn.net/10km/article/details/68926498

猜你喜欢

转载自www.cnblogs.com/lwjkz/p/11481888.html
今日推荐