predict: use smaller image for face detect

koush · Jun 5, 2024 · 2a2f96a · 2a2f96a
1 parent dc9b5f4
commit 2a2f96a
Show file tree

Hide file tree

Showing 9 changed files with 15 additions and 15 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.57"
+   "version": "0.1.58"
 }
diff --git a/plugins/coreml/src/coreml/face_recognition.py b/plugins/coreml/src/coreml/face_recognition.py
@@ -71,7 +71,7 @@ async def predictFaceModel(self, input: np.ndarray):
         def predict():
             model, inputName = self.faceModel
             out_dict = model.predict({inputName: input})
-            results = out_dict["var_2167"][0]
+            results = list(out_dict.values())[0][0]
             return results
         results = await asyncio.get_event_loop().run_in_executor(
             self.recogExecutor, lambda: predict()

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.95"
+   "version": "0.1.96"
 }
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.96"
+   "version": "0.1.97"
 }
diff --git a/plugins/tensorflow-lite/src/predict/face_recognize.py b/plugins/tensorflow-lite/src/predict/face_recognize.py
@@ -20,16 +20,16 @@ class FaceRecognizeDetection(PredictPlugin):
     def __init__(self, nativeId: str | None = None):
         super().__init__(nativeId=nativeId)
 
-        self.inputheight = 640
-        self.inputwidth = 640
+        self.inputheight = 320
+        self.inputwidth = 320
 
         self.labels = {
             0: "face",
         }
         self.loop = asyncio.get_event_loop()
         self.minThreshold = 0.2
 
-        self.detectModel = self.downloadModel("scrypted_yolov10n_face")
+        self.detectModel = self.downloadModel("scrypted_yolov10n_face_320")
         self.faceModel = self.downloadModel("inception_resnet_v1")
 
     def downloadModel(self, model: str):

diff --git a/server/package-lock.json b/server/package-lock.json