首先需要注意的是,要把demo.py需要检测的图片放到data/demo文件夹下,才能够使之检测出来!!!!!直接把所有测试图像copy到py-faster-rcnn/data/demo文件夹下即可。
在上面说的做好后,就可以开始下面的更改了:
for im_name in im_names:
print '~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~'
print 'Demo for data/demo/{}'.format(im_name)
demo(net, im_name)
plt.savefig("../testfigs/" + im_name)
#plt.show()
其实就是增加一句:
plt.savefig("../testfigs/" + im_name)
注销最后一句图片显示部分:
#plt.show()