Faster-RCNN训练问题解决

来源：互联网发布：mg汽车知乎编辑：程序博客网时间：2024/05/24 04:16

'RPN_POST_NMS_TOP_N': 2000,           'RPN_PRE_NMS_TOP_N': 12000,           'SCALES': [600],           'SNAPSHOT_INFIX': 'stage1',           'SNAPSHOT_ITERS': 10000,           'USE_FLIPPED': True,           'USE_PREFETCH': False}, 'USE_GPU_NMS': True}Loaded dataset `voc_2007_trainval` for trainingSet proposal method: gtAppending horizontally-flipped training examples...Process Process-1:Traceback (most recent call last):  File "/usr/lib/python2.7/multiprocessing/process.py", line 258, in _bootstrap    self.run()  File "/usr/lib/python2.7/multiprocessing/process.py", line 114, in run    self._target(*self._args, **self._kwargs)  File "./tools/train_faster_rcnn_alt_opt.py", line 122, in train_rpn    roidb, imdb = get_roidb(imdb_name)  File "./tools/train_faster_rcnn_alt_opt.py", line 67, in get_roidb    roidb = get_training_roidb(imdb)  File "/home/py-faster-rcnn/tools/../lib/fast_rcnn/train.py", line 118, in get_training_roidb    imdb.append_flipped_images()  File "/home/py-faster-rcnn/tools/../lib/datasets/imdb.py", line 106, in append_flipped_images    boxes = self.roidb[i]['boxes'].copy()  File "/home/py-faster-rcnn/tools/../lib/datasets/imdb.py", line 67, in roidb    self._roidb = self.roidb_handler()  File "/home/py-faster-rcnn/tools/../lib/datasets/pascal_voc.py", line 103, in gt_roidb    for index in self.image_index]  File "/home/py-faster-rcnn/tools/../lib/datasets/pascal_voc.py", line 208, in _load_pascal_annotation    cls = self._class_to_ind[obj.find('name').text.lower().strip()]KeyError: 'lab'

append_flipped_images()添加镜像roi

解决方法：

import pdb
......

print()
pdb.set_trace()

阅读全文

0 0