We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3e171db commit 8e029c5Copy full SHA for 8e029c5
lib/roi_data_layer/minibatch.py
@@ -40,7 +40,7 @@ def get_minibatch(roidb, num_classes):
40
gt_inds = np.where(roidb[0]['gt_classes'] != 0)[0]
41
else:
42
# For the COCO ground truth boxes, exclude the ones that are ''iscrowd''
43
- gt_inds = np.where(roidb[0]['gt_classes'] != 0 & np.all(roidb[0]['gt_overlaps'].toarray() > -1.0, axis=1))[0]
+ gt_inds = np.where((roidb[0]['gt_classes'] != 0) & np.all(roidb[0]['gt_overlaps'].toarray() > -1.0, axis=1))[0]
44
gt_boxes = np.empty((len(gt_inds), 5), dtype=np.float32)
45
gt_boxes[:, 0:4] = roidb[0]['boxes'][gt_inds, :] * im_scales[0]
46
gt_boxes[:, 4] = roidb[0]['gt_classes'][gt_inds]
0 commit comments