训练faster rcnn出现AssertionError的解决办法

在使用caffe训练faster rcnn的时候，报了一个断言错误：

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

问题解决：

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减一去掉
将对Xmin,Ymin,Xmax,Ymax减一去掉

 

3、（可选，如果1和2可以解决问题，就没必要用3）修改lib/fast_rcnn/config.py，不使图片实现翻转，如下改为

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

 

问题解决！

 

参照博客：点击打开链接