windows 下faster rcnn入门2:根据需求更改


前提基于 https://github.com/dBeker/Faster-RCNN-TensorFlow-Python3.5代码,网上有两版不同的tensorflow下的faster rcnn代码

1、检测标记所有类别:

faster rcnn是对每一个图片进行分类别测试,有点太乱。参照faster rcnn demo.py:在一个窗口显示所有类别标注,进行修改。

2、保存目标检测结果

demo.py 倒数第二行添加(基本参数去除空白)
plt.savefig("./data/demo_result/" + im_name, format = 'jpg',transparent = True,pad_inches = 0,dpi = 300,bbox_inches = 'tight') #add

3、计算map、ap

修改demo.py,(参照博客
(1)在开头,import这里加入

from lib.utils.test import test_net
from lib.datasets.factory import get_imdb

(2)最后加入:(注意与plt.show()对齐)

    imdb = get_imdb("voc_2007_trainval")
    test_net(sess, net, imdb, 'default')

4、测试数据处理

A:不想改测试文件默认路径:采取了一个方法,把所有测试文件名读入到一个txt,然后demo.py使用时再把文件名读入到一个list中。

1、新建一个writetxt.py文件
在这里插入图片描述
2、demo.py修改:
1)新建函数
在这里插入图片描述
2)main函数修改:
在这里插入图片描述
B:后来又发现博客上有一个简单方法,设置路径:(还是python没学好)

im_names = os.listdir(cfg.FLAGS2["data_dir"]+'/demo') #测试图片所在位置
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值