tensorflow-lite: switch usb edgetpu model

koush · Mar 5, 2024 · 6ebab81 · 6ebab81
1 parent 9d92154
commit 6ebab81
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 6 deletions.
diff --git a/plugins/tensorflow-lite/.vscode/settings.json b/plugins/tensorflow-lite/.vscode/settings.json
@@ -8,9 +8,13 @@
     // "scrypted.debugHost": "192.168.2.119",
     // "scrypted.serverRoot": "/home/pi/.scrypted",
 
+    // lxc installation
+    "scrypted.debugHost": "scrypted-server",
+    "scrypted.serverRoot": "/root/.scrypted",
+
     // local checkout
-    "scrypted.debugHost": "127.0.0.1",
-    "scrypted.serverRoot": "/Users/koush/.scrypted",
+    // "scrypted.debugHost": "127.0.0.1",
+    // "scrypted.serverRoot": "/Users/koush/.scrypted",
     // "scrypted.debugHost": "koushik-windows",
     // "scrypted.serverRoot": "C:\\Users\\koush\\.scrypted",
 

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.47"
+   "version": "0.1.48"
 }
diff --git a/plugins/tensorflow-lite/src/tflite/__init__.py b/plugins/tensorflow-lite/src/tflite/__init__.py
@@ -70,7 +70,13 @@ def configureModel():
             nonlocal model
 
             if defaultModel:
-                model = "efficientdet_lite0_320_ptq"
+                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"
+                else:
+                    model = "efficientdet_lite0_320_ptq"
             self.yolo = "yolo" in model
             self.yolov8 = "yolov8" in model