We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 5638fc7 + f221eb4 commit 368115aCopy full SHA for 368115a
lib/roi_data_layer/minibatch.py
@@ -46,7 +46,7 @@ def get_minibatch(roidb, num_classes):
46
gt_boxes[:, 4] = roidb[0]['gt_classes'][gt_inds]
47
blobs['gt_boxes'] = gt_boxes
48
blobs['im_info'] = np.array(
49
- [[im_blob.shape[1], im_blob.shape[2], im_scales[0]]],
+ [im_blob.shape[1], im_blob.shape[2], im_scales[0]],
50
dtype=np.float32)
51
52
return blobs
0 commit comments