faster r-cnn训练问题集

来源：互联网发布：男学生衬衣知乎编辑：程序博客网时间：2024/05/16 03:16

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

问题分析：检查自己数据发现，左上角坐标（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