Merge branch 'master' of https://github.com/endernewton/tf-faster-rcnn

study3h · Aug 14, 2017 · 368115a · 368115a
2 parents 5638fc7 + f221eb4
commit 368115a
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