Faster RCNN修改demo.py文件实现图片的批量测试与保存

关于Faster R-CNN Tensorflow+python 3.5 在Windows10环境下配置实现,可以参看这里。运行在demo.py文件中测试数据中源码设置仅检测几张图片供参考,原始的代码段如下。

im_names = ['000456.jpg', '000457.jpg', '000542.jpg', '001150.jpg',
                '001763.jpg', '004545.jpg']
    for im_name in im_names:
        print('~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~')
        print('Demo for data/demo/{}'.format(im_name))
        demo(sess, net, im_name)

    plt.show()

问题1:一旦测试图片多起来,桌面上一下子显示出十几个figure窗口看起来有点小刺激,所以可以考虑修改以下代码,进行批量测试和保存。

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

 im_names = os.listdir(cfg.FLAGS2["data_dir"]+'/dem
  • 4
    点赞
  • 31
    收藏
    觉得还不错? 一键收藏
  • 11
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 11
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值