No public description

PiperOrigin-RevId: 898054992

Author: tensorflower-gardener

official/projects/waste_identification_ml/model_inference/cn_model_run.ipynb

@@ -111,7 +111,7 @@
       },
       "cell_type": "code",
       "source": [
-        "!wget https://storage.googleapis.com/tf_model_garden/vision/waste_identification_ml/CN-ModelCheckpoints/checkpoint_best_total_432x432.pth"
+        "!wget https://storage.googleapis.com/tf_model_garden/vision/waste_identification_ml/CN-ModelCheckpoints/ModelRegistry_432x432_March26/checkpoint_best_total.pth"
       ],
       "outputs": [],
       "execution_count": null
@@ -122,7 +122,7 @@
       },
       "cell_type": "code",
       "source": [
-        "seg_medium_model = RFDETRSegMedium(pretrain_weights=\"./checkpoint_best_total_432x432.pth\")\n",
+        "seg_medium_model = RFDETRSegMedium(pretrain_weights=\"./checkpoint_best_total.pth\")\n",
         "class_id_to_class_name_mapper = seg_medium_model.class_names"
       ],
       "outputs": [],
@@ -302,9 +302,9 @@
       },
       "cell_type": "code",
       "source": [
-        "img_path = \"./sample_image.jpg\"\n",
+        "input_image_path = \"./sample_image.jpg\"\n",
         "onnx_model_path = \"./inference_model.onnx\"\n",
-        "output_image_dimensions = (1024, 1024)"
+        "output_image_dimension = (1920, 1080)"
       ],
       "outputs": [],
       "execution_count": null
@@ -345,8 +345,8 @@
       },
       "cell_type": "code",
       "source": [
-        "image_array = model_utils._get_input_batch_for_inference(image_path=img_path)\n",
-        "outputs = onnx_model_session.run(None, {\"input\": image_array})"
+        "image_array = model_utils._get_input_batch_for_inference(image_path=input_image_path)\n",
+        "outputs = onnx_model.run(None, {\"input\": image_array})"
       ],
       "outputs": [],
       "execution_count": null
@@ -362,17 +362,20 @@
     },
     {
       "metadata": {
-        "id": "POQajqV7IeGZ"
+        "id": "6eucX1xOsyrr"
       },
-      "cell_type": "markdown",
+      "cell_type": "code",
       "source": [
         "results = model_utils._reformat_triton_output_to_dict(\n",
         "    outputs, confidence_threshold=0.5, max_boxes=100\n",
         ")\n",
         "\n",
-        "results = model_utils._scale_bbox_and_masks(results, target_dims=(1920, 1080))\n",
+        "# output_image_dimension makes sure the output image dimession along with the bounding boxes and masks are resized.\n",
+        "results = model_utils._scale_bbox_and_masks(results, target_dims=output_image_dimension)\n",
         "results[\"class_names\"] = model_utils.get_class_names(results)"
-      ]
+      ],
+      "outputs": [],
+      "execution_count": null
     },
     {
       "metadata": {
@@ -385,13 +388,15 @@
     },
     {
       "metadata": {
-        "id": "dwHzAAwjIjbu"
+        "id": "4tIuPGpls0h4"
       },
-      "cell_type": "markdown",
+      "cell_type": "code",
       "source": [
-        "draw_detections_and_save_image(img=Image.open(img_path), results=results, save_path= \"./output.jpg\")\n",
+        "draw_detections_and_save_image(img=Image.open(input_image_path), results=results, save_path= \"./output.jpg\")\n",
         "Image.open(\"./output.jpg\")"
-      ]
+      ],
+      "outputs": [],
+      "execution_count": null
     },
     {
       "metadata": {