TFFRCNN icon indicating copy to clipboard operation
TFFRCNN copied to clipboard

KeyError: 'max_overlaps'

Open Esugi8 opened this issue 8 years ago • 3 comments

I got ResourceExhaustedError when running train_net.py with original PASCAL VOC 2007, so I reduced the number of Images and Annotations to 1000. Then I tried to run train_net.py again, and got an error below.

What should I do for solving this error? Could you please let me know any solution for this?

Computing bounding-box regression targets... Traceback (most recent call last): File "./faster_rcnn/train_net.py", line 109, in restore=bool(int(args.restore))) File "./faster_rcnn/../lib/fast_rcnn/train.py", line 394, in train_net sw = SolverWrapper(sess, network, imdb, roidb, output_dir, logdir= log_dir, pretrained_model=pretrained_model) File "./faster_rcnn/../lib/fast_rcnn/train.py", line 44, in init self.bbox_means, self.bbox_stds = rdl_roidb.add_bbox_regression_targets(roidb) File "./faster_rcnn/../lib/roi_data_layer/roidb.py", line 65, in add_bbox_regression_targets max_overlaps = roidb[im_i]['max_overlaps'] KeyError: 'max_overlaps'

Esugi8 avatar Jan 30 '17 05:01 Esugi8

Hi, @Esugi8 Did you figure out this problem? ResourceExhaustedError sounds like your GPU does not have enough memories. You can try to set TRAIN.SCALES to a small number, like 400.

CharlesShang avatar Feb 23 '17 12:02 CharlesShang

did you solve it?

liu09114 avatar Apr 13 '17 09:04 liu09114

try clean up the roidb cache file in $TFFRCNN/data/cache. It works for me to solve the 'max_overlaps' problem.

AuroraLHT avatar Jun 16 '17 04:06 AuroraLHT