python - 1 つのクラスを使用した VOC データセットでの py_faster_rcnn のトレーニング

Question

私は py_faster_rcnn を使用して、1 つのクラス (「人」) のシステムをトレーニングしています。もともと、この投稿 How to train new fast-rcnn imagesset のようなアサーション エラーが表示されました。

そのため、imdb.py ファイルに次の変更を加えました。

for b in range(len(boxes)):
    if boxes[b][2] < boxes[b][0]:
       boxes[b][0] = 0
assert (boxes[:,2] >= boxes[:,0]).all()

上記の変更後、この新しいエラーが発生します。誰かがこのエラーに遭遇したことがありますか、それとも私が間違っている可能性がありますか?

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 130, in train_rpn
max_iters=max_iters)
File "/home/microway/test/pytest/py-faster-rcnn/tools/../lib/fast_rcnn/train.py", line 134, in train_net
pretrained_model=pretrained_model)
File "/home/microway/test/pytest/py-faster-rcnn/tools/../lib/fast_rcnn/train.py", line 53, in __init__
self.solver.net.layers[0].set_roidb(roidb)
File "/home/microway/test/pytest/py-faster-rcnn/tools/../lib/roi_data_layer/layer.py", line 68, in set_roidb
self._shuffle_roidb_inds()
File "/home/microway/test/pytest/py-faster-rcnn/tools/../lib/roi_data_layer/layer.py", line 26, in _shuffle_roidb_inds
widths = np.array([r['width'] for r in self._roidb])
KeyError: 'width'