Add testing trained model on sample image and lite2 model

jayjunlee · Sep 19, 2021 · b05cdcf · b05cdcf
1 parent fdb213d
commit b05cdcf
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/cnn_ball_detection.ipynb b/cnn_ball_detection.ipynb
@@ -757,6 +757,7 @@
    "execution_count": null,
    "source": [
     "spec = model_spec.get('efficientdet_lite0')\n",
+    "#spec = model_spec.get('efficientdet_lite2')\n",
     "train_data, validation_data, test_data = object_detector.DataLoader.from_csv('./dataset/label/labels_tflite.csv')\n",
     "model = object_detector.create(train_data, model_spec=spec, batch_size=16, train_whole_model=True, validation_data=validation_data)"
    ],
@@ -883,6 +884,18 @@
    ],
    "outputs": [],
    "metadata": {}
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "source": [
+    "interpreter = tf.lite.Interpreter(model_path=model_path)\n",
+    "interpreter.allocate_tensors()\n",
+    "detection_result_image = run_odt_and_draw_results(\".dataset/processed/image_blue_bright_25.jpg\", interpreter, threshold=0.3)\n",
+    "Image.fromarray(detection_result_image)"
+   ],
+   "outputs": [],
+   "metadata": {}
   }
  ],
  "metadata": {

diff --git a/trained_model/efficientdet_lite2.tflite b/trained_model/efficientdet_lite2.tflite