解决faster-rcnn中训练时assert(boxes[:,2]>=boxes[:,0]).all()的问题

来源:互联网 发布:走好网络群众路线 编辑:程序博客网 时间:2024/04/28 05:14

问题:在用py-faster-rcnn进行训练时出现 “assert(boxes[:,2]>=boxes[:,0]).all()“ 报错提示

原因:目标框坐标溢出范围,如标注的目标从左上角开始即坐标可能为0,而在源代码中会对坐标进行减1,导致出错


解决方法:

方法1:

a::修改lib/datasets/imdb.py,append_flipped_images()函数
数据整理,在一行代码为 boxes[:, 2] = widths[i] - oldx1 - 1下加入代码:
for b in range(len(boxes)):
if boxes[b][2]< boxes[b][0]:
boxes[b][0] = 0b::
修改lib/datasets/pascal_voc.py,_load_pascal_annotation(,)函数
对Xmin,Ymin,Xmax,Ymax减一去掉,变为:

方法2:

修改lib/fast_rcnn/config.py

# Use horizontally-flipped images during training?__C.TRAIN.USE_FLIPPED = False

阅读全文
0 0
原创粉丝点击