predict: relax face threshold

koush · Jun 4, 2024 · 2f19866 · 2f19866
1 parent 9a770e9
commit 2f19866
Show file tree

Hide file tree

Showing 7 changed files with 11 additions and 13 deletions.
diff --git a/plugins/coreml/package-lock.json b/plugins/coreml/package-lock.json
diff --git a/plugins/coreml/package.json b/plugins/coreml/package.json
@@ -42,5 +42,5 @@
    "devDependencies": {
       "@scrypted/sdk": "file:../../sdk"
    },
-   "version": "0.1.56"
+   "version": "0.1.57"
 }
diff --git a/plugins/onnx/package-lock.json b/plugins/onnx/package-lock.json
diff --git a/plugins/onnx/package.json b/plugins/onnx/package.json
@@ -42,5 +42,5 @@
    "devDependencies": {
       "@scrypted/sdk": "file:../../sdk"
    },
-   "version": "0.1.94"
+   "version": "0.1.95"
 }
diff --git a/plugins/openvino/package-lock.json b/plugins/openvino/package-lock.json
diff --git a/plugins/openvino/package.json b/plugins/openvino/package.json
@@ -42,5 +42,5 @@
    "devDependencies": {
       "@scrypted/sdk": "file:../../sdk"
    },
-   "version": "0.1.95"
+   "version": "0.1.96"
 }
diff --git a/plugins/tensorflow-lite/src/predict/face_recognize.py b/plugins/tensorflow-lite/src/predict/face_recognize.py
@@ -25,13 +25,11 @@ def __init__(self, nativeId: str | None = None):
 
         self.labels = {
             0: "face",
-            1: "plate",
-            2: "text",
         }
         self.loop = asyncio.get_event_loop()
-        self.minThreshold = 0.7
+        self.minThreshold = 0.2
 
-        self.detectModel = self.downloadModel("scrypted_yolov10m_face")
+        self.detectModel = self.downloadModel("scrypted_yolov10n_face")
         self.faceModel = self.downloadModel("inception_resnet_v1")
 
     def downloadModel(self, model: str):