在faster-rcnn中用wider数据库是进行的改动

最新推荐文章于 2021-03-19 06:15:47 发布

annw0922

最新推荐文章于 2021-03-19 06:15:47 发布

阅读量362

点赞数

本文链接：https://blog.csdn.net/forteenscoops/article/details/79738870

版权

在应用faster-rcnn模型并使用wider数据库时，作者遇到了与pascal_voc.py文件相关的问题。通过注释掉与'difficult'相关的内容解决了困难物体的处理。随后在imdb.py文件中遇到assertion错误，由于数据集中所有框都只有一个face，不满足原有条件。作者找到了解决方案，并对voc_eval内的代码进行了相应修改。

摘要由CSDN通过智能技术生成

File "/home/wangyj/py-faster-rcnn/tools/../lib/datasets/pascal_voc.py", line 192, in _load_pascal_annotation
obj for obj in objs if int(obj.find('difficult').text) == 0]

pascal_voc.py的line 188起注释掉了

if not self.config['use_diff']:

# Exclude the samples labeled as difficult
non_diff_objs = [
obj for obj in objs if int(obj.find('difficult').text) == 0]
# if len(non_diff_objs) != len(objs):
# print 'Removed {} difficult objects'.format(
# len(objs) - len(non_diff_objs))
objs = non_diff_objs

在voc_eval中注释掉了