From 2f19866f051c525a99e41b9f13b4ba34868389ac Mon Sep 17 00:00:00 2001 From: Koushik Dutta Date: Mon, 3 Jun 2024 23:34:01 -0700 Subject: [PATCH] predict: relax face threshold --- plugins/coreml/package-lock.json | 4 ++-- plugins/coreml/package.json | 2 +- plugins/onnx/package-lock.json | 4 ++-- plugins/onnx/package.json | 2 +- plugins/openvino/package-lock.json | 4 ++-- plugins/openvino/package.json | 2 +- plugins/tensorflow-lite/src/predict/face_recognize.py | 6 ++---- 7 files changed, 11 insertions(+), 13 deletions(-) diff --git a/plugins/coreml/package-lock.json b/plugins/coreml/package-lock.json index 78bcff66a7..cd18c80183 100644 --- a/plugins/coreml/package-lock.json +++ b/plugins/coreml/package-lock.json @@ -1,12 +1,12 @@ { "name": "@scrypted/coreml", - "version": "0.1.56", + "version": "0.1.57", "lockfileVersion": 2, "requires": true, "packages": { "": { "name": "@scrypted/coreml", - "version": "0.1.56", + "version": "0.1.57", "devDependencies": { "@scrypted/sdk": "file:../../sdk" } diff --git a/plugins/coreml/package.json b/plugins/coreml/package.json index 86e638bef4..fd4c0afb0a 100644 --- 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 index d6daac7ffc..1c1440ce7e 100644 --- a/plugins/onnx/package-lock.json +++ b/plugins/onnx/package-lock.json @@ -1,12 +1,12 @@ { "name": "@scrypted/openvino", - "version": "0.1.94", + "version": "0.1.95", "lockfileVersion": 2, "requires": true, "packages": { "": { "name": "@scrypted/openvino", - "version": "0.1.94", + "version": "0.1.95", "devDependencies": { "@scrypted/sdk": "file:../../sdk" } diff --git a/plugins/onnx/package.json b/plugins/onnx/package.json index e58124584e..44d77f1bf2 100644 --- 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 index 1c1440ce7e..dc7e0ba9e9 100644 --- a/plugins/openvino/package-lock.json +++ b/plugins/openvino/package-lock.json @@ -1,12 +1,12 @@ { "name": "@scrypted/openvino", - "version": "0.1.95", + "version": "0.1.96", "lockfileVersion": 2, "requires": true, "packages": { "": { "name": "@scrypted/openvino", - "version": "0.1.95", + "version": "0.1.96", "devDependencies": { "@scrypted/sdk": "file:../../sdk" } diff --git a/plugins/openvino/package.json b/plugins/openvino/package.json index 84d0357f00..e7c08e4f9b 100644 --- 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 index a8b3801774..f2cdb4202c 100644 --- 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):