caffe下使用faster-rcnn批量测试部分图片及结果保存

caffe下使用faster-rcnn批量测试部分图片及结果保存

找到修改demo.py得到的测试文件demo_mine.py。主要修改下文件的保存路径与输出格式,因为不少测试图片经plt.savefig函数保存,如果不加其他参数设置会导致图像上存在不少空白位置,plt.savefig函数的详细参数设置可以参考官方文档:

   im_names = os.listdir('E:\py-faster-rcnn-master\data\demo') #测试图片所在位置
   for im_name in im_names:
       print '~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~'
       print 'Demo for data/demo/{}'.format(im_name)
       demo(net, im_name)
       # 保存测试图片所在位置,并设置输出格式
       plt.savefig('E:\py-faster-rcnn-master\data\save_demo_test/’ +  im_name,
                   format = 'png', transparent = True, pad_inches = 0,
                   dpi = 300, bbox_inches = 'tight')

#   plt.show()

将最后一行注释掉防止测试结果图片一下子全部跳出,save_demo_test是自己建的保存结果文件夹。
cmd到E:\py-faster-rcnn-master目录下输入“python demo_mine.py”。

https://blog.csdn.net/jcli1_14/article/details/82788170
https://blog.csdn.net/10km/article/details/68926498

  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值