diff --git a/lib/roi_data_layer/roidb.py b/lib/roi_data_layer/roidb.py index 97a6a761..dd857e72 100644 --- a/lib/roi_data_layer/roidb.py +++ b/lib/roi_data_layer/roidb.py @@ -29,6 +29,7 @@ def prepare_roidb(imdb): roidb[i]['height'] = sizes[i][1] # need gt_overlaps as a dense array for argmax gt_overlaps = roidb[i]['gt_overlaps'].toarray() + # # max overlap with gt over classes (columns) max_overlaps = gt_overlaps.max(axis=1) # gt class that had the max overlap