faster-rcnn训练遇到的问题

秒速五厘米 2022-05-15 05:38 194阅读 0赞

## 1. 问题1 ##

_sample_rois(roidb, fg_rois_per_image, rois_per_image, num_classes)

报错是因为是$FRCN\_ROOT/lib/roi\_data\_layer下的minibatch.py中的npr.choice引起的，解决方法：  
修改：

if fg_inds.size > 0:
            for i in range(0,len(fg_inds)):
                fg_inds[i] = int(fg_inds[i])
            fg_inds = npr.choice(fg_inds, size=int(fg_rois_per_this_image), replace=False)

**注意：**在这里是有两处使用了npr.choice函数所以需要fg和bg两处

## 2. 问题2 ##

在

labels[fg_rois_per_this_image:] = 0
    TypeError: slice indices must be integers or None or have an index method

解决方法：  
报错是因为这个错误是由numpy的版本引起的，只要将fg\_rois\_per\_this\_image强制转换为int型就可以了

labels[int(fg_rois_per_this_image):] = 0

## 3. 问题3 ##

训练faster-rcnn出现错误“KeyError: ‘max\_overlaps’”  
解决方法：  
删除py-faster-rcnn/data/VOCdevkit2007/annotations\_cache这个文件夹；  
删除py-faster-rcnn/data/cache文件夹。

## 4. 问题4 ##

出现anchor\_target\_layer.py”, line 137, in forward gt\_argmax\_overlaps = overlaps.argmax(axis=0)错误  
解决方法：产生这个错误的原因是因为输入图片的长度不满足要求，训练faster-rcnn需要图片的长宽比(width/height)要在0.462-6.828之间，就是太过瘦长的图片不要

## 5. 问题5 ##

出现错误

protobuf'module' object has no attribute 'text_format

是因为protobuf的版本发生了变化，  
解决方法：在文件./lib/fast\_rcnn/train.py增加一行import google.protobuf.text\_format 即可解决问题

## 6. 问题6 ##

datasets/imdb.py”, line 108, in append\_flipped\_images assert (boxes\[:, 2\] >= boxes\[:, 0\]).all()  
问题分析：  
检查自己数据发现，左上角坐标（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减一去掉，变为：

# Make pixel indexes 0-based
    x1 = float(bbox.find('xmin').text)  # - 1
    y1 = float(bbox.find('ymin').text)  # - 1
    x2 = float(bbox.find('xmax').text)  # - 1
    y2 = float(bbox.find('ymax').text)  # - 1

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

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

## 7. 问题7 ##

错误信息：cudnn.hpp:127:41: error: too few arguments to function ‘cudnnStatus\_t cudnnSetPooling2dDescriptor  
这是因为当前版本的caffe的cudnn实现与系统所安装的cudnn的版本不一致引起的。

解决办法：  
1.将./include/caffe/util/cudnn.hpp 换成最新版的caffe里的cudnn的实现，即相应的cudnn.hpp.  
2. 将./include/caffe/layers里的，所有以cudnn开头的文件，例如cudnn\_conv\_layer.hpp。 都替换成最新版的caffe里的相应的同名文件。  
3.将./src/caffe/layers里的，所有以cudnn开头的文件，例如cudnn\_lrn\_layer.cu，cudnn\_pooling\_layer.cpp，cudnn\_sigmoid\_layer.cu。  
都替换成最新版的caffe里的相应的同名文件。