Fixed bbox scaling to be applied to all boxes.

oztc · Dec 13, 2017 · 850821a · 850821a
1 parent 9a89cd0
commit 850821a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/ResNet50RetinaNet - COCO 2017.ipynb b/examples/ResNet50RetinaNet - COCO 2017.ipynb
@@ -152,7 +152,7 @@
     "scores = detections[0, np.arange(detections.shape[1]), 4 + predicted_labels]\n",
     "\n",
     "# correct for image scale\n",
-    "detections[:, :4] /= scale\n",
+    "detections[0, :, :4] /= scale\n",
     "\n",
     "# visualize detections\n",
     "for idx, (label, score) in enumerate(zip(predicted_labels, scores)):\n",