Faster-RCNN训练问题解决

最新推荐文章于 2023-05-21 23:39:11 发布

cv.exp

最新推荐文章于 2023-05-21 23:39:11 发布

阅读量794

点赞数

分类专栏： Deep Learning

本文链接：https://blog.csdn.net/forest_world/article/details/78149101

版权

Deep Learning 专栏收录该内容

172 篇文章 6 订阅

订阅专栏

'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 training
Set proposal method: gt
Appending 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()