Caffe各种Bug修正
如何运行
-
test
./tools/test_net.py --gpu 0 --def models/pascal_voc/VGG16/faster_rcnn_end2end/test.prototxt --net /home/drl/project/py-faster-rcnn/data/faster_rcnn_models/vgg16_faster_rcnn_iter_90000.caffemodel --imdb voc_2007_test --cfg experiments/cfgs/faster_rcnn_end2end.yml -
train
./experiments/scripts/faster_rcnn_end2end.sh 0 VGG16 pascal_voc -
train
./tools/train_net.py --gpu 0 --solver models/pascal_voc/VGG16/faster_rcnn_end2end/solver.prototxt --imdb voc_2007_trainval --iters 50 --cfg experiments/cfgs/faster_rcnn_end2end.yml -
train
./experiments/scripts/faster_rcnn_end2end.sh 0 VGG16 pascal_voc
Bug汇总
-
AttributeError: ‘module’ object has no attribute ‘text_format’
-
bbox_targets[ind, start:end] = bbox_target_data[ind, 1:]
TypeError: slice indices must be integers or None or have an index method
An easier solution is to add the following lines to lib/proposal_target_layer.py
After line 126,
start=int(start)
end=int(end)
After line 166,
fg_rois_per_this_image=int(fg_rois_per_this_image)
-
BB = BB[sorted_ind, :]
IndexError: too many indices for array -
修改test.protxt和train.protxt的cls_score和bbox_pred 为_1
修改test.py bbox_pred为bbox_pred_1
修改voc_eval.py int变成float
-
VGG16.V2.CAFFEMODEL
-
assert (boxes[:, 2] >= boxes[:, 0]).all()
AssertionError
自己制作了VOC或者coco数据集格式
如果你自己制作了voc pascal或者coco数据集格式,那么你需要注意,看看是否有类似下面的报错
RuntimeWarning: invalid value encountered in log targets_dw = np.log(gt_widths / ex_widths)
这种报错说明数据集的数据有一些问题,多出现在没有控制好边界的情况,首先,打开lib/database/pascal_voc.py文件,找到208行,将208行至211行每一行后面的-1删除,如下所示:
x1 = float(bbox.find(‘xmin’).text)
y1 = float(bbox.find(‘ymin’).text)
x2 = float(bbox.find(‘xmax’).text)
y2 = float(bbox.find(‘ymax’).text)
原因是因为我们制作的xml文件中有些框的坐标是从左上角开始的,也就是(0,0)如果再减一就会出现log(-1)的情况
如果这样之后还是出现类似的报错,那么说明依然有-1或者其他负数的情况出现。解决方法是打开./lib/model/config.py文件,找到flipp选项,将其置为False
__C.TRAIN.USE_FLIPPED = False
如果这样以后还是报类似的错误,就一定检查一下自己制作数据集的过程,看看是否哪里没有考虑清
作者:ZeroZone零域
来源:CSDN
原文:https://blog.csdn.net/ksws0292756/article/details/80702704
版权声明:本文为博主原创文章,转载请附上博文链接!