Update minibatch.py

snxmv · Aug 14, 2017 · f221eb4 · f221eb4
1 parent b973042
commit f221eb4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/roi_data_layer/minibatch.py b/lib/roi_data_layer/minibatch.py
@@ -46,7 +46,7 @@ def get_minibatch(roidb, num_classes):
   gt_boxes[:, 4] = roidb[0]['gt_classes'][gt_inds]
   blobs['gt_boxes'] = gt_boxes
   blobs['im_info'] = np.array(
-    [[im_blob.shape[1], im_blob.shape[2], im_scales[0]]],
+    [im_blob.shape[1], im_blob.shape[2], im_scales[0]],
     dtype=np.float32)
 
   return blobs