Simpler bounding box

Author: mattdangerw

chapter12_object-detection.ipynb

@@ -602,6 +602,7 @@
    "source": [
     "detector = keras_hub.models.ObjectDetector.from_preset(\n",
     "    \"retinanet_resnet50_fpn_v2_coco\",\n",
+    "    bounding_box_format=\"rel_xywh\",\n",
     ")\n",
     "predictions = detector.predict(image)"
    ]
@@ -639,8 +640,7 @@
     "ig, ax = plt.subplots(dpi=300)\n",
     "draw_image(ax, path)\n",
     "for i in range(predictions[\"num_detections\"][0]):\n",
-    "    y1, x1, y2, x2 = predictions[\"boxes\"][0][i]\n",
-    "    box = (x1 / 800, y1 / 800, (x2 - x1) / 800, (y2 - y1) / 800)\n",
+    "    box = predictions[\"boxes\"][0][i]\n",
     "    label = predictions[\"labels\"][0][i]\n",
     "    label_name = keras_hub.utils.coco_id_to_name(label)\n",
     "    draw_box(ax, box, label_name, label_to_color(label))\n",