tensorflow-lite: change default model for usb

koush · Apr 7, 2024 · 96d536f · 96d536f
1 parent c678b31
commit 96d536f
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 7 deletions.
diff --git a/plugins/tensorflow-lite/package-lock.json b/plugins/tensorflow-lite/package-lock.json
diff --git a/plugins/tensorflow-lite/package.json b/plugins/tensorflow-lite/package.json
@@ -53,5 +53,5 @@
    "devDependencies": {
       "@scrypted/sdk": "file:../../sdk"
    },
-   "version": "0.1.50"
+   "version": "0.1.51"
 }
diff --git a/plugins/tensorflow-lite/src/tflite/__init__.py b/plugins/tensorflow-lite/src/tflite/__init__.py
@@ -71,10 +71,7 @@ def configureModel():
 
             if defaultModel:
                 if edge_tpus and next((obj for obj in edge_tpus if obj['type'] == 'usb'), None):
-                    model = "yolov8n_full_integer_quant_320"
-                    # this model seems completely wacky with lots of false positives.
-                    # might be broken?
-                    # model = "ssdlite_mobiledet_coco_qat_postprocess"
+                    model = "ssdlite_mobiledet_coco_qat_postprocess"
                 else:
                     model = "efficientdet_lite0_320_ptq"
             self.yolo = "yolo" in model