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

来源：互联网发布：百度笑报程序员编辑：程序博客网时间：2024/04/28 17:50

1、出现问题：训练faster rcnn时出现如下报错：

File "/py-faster-rcnn/tools/../lib/datasets/imdb.py", line 108, in append_flipped_images    assert (boxes[:, 2] >= boxes[:, 0]).all()AssertionError

2、问题分析：

检查自己数据发现，左上角坐标（x,y）可能为0，或标定区域溢出图片

而faster rcnn会对Xmin,Ymin,Xmax,Ymax进行减一操作如果Xmin为0，减一后变为65535

问题解决1、修改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] = 0

2、修改lib/datasets/pascal_voc.py，_load_pascal_annotation(,)函数将对Xmin,Ymin,Xmax,Ymax减一去掉，变为：
3、（可选，如果1和2可以解决问题，就没必要用3）修改lib/fast_rcnn/config.py，不使图片实现翻转，如下改为：
# Use horizontally-flipped images during training?__C.TRAIN.USE_FLIPPED = False
问题解决

阅读全文

0 0