diff --git a/plugins/ncnn/package-lock.json b/plugins/ncnn/package-lock.json index 8d9455e74..311ed9bf9 100644 --- a/plugins/ncnn/package-lock.json +++ b/plugins/ncnn/package-lock.json @@ -1,12 +1,12 @@ { "name": "@scrypted/ncnn", - "version": "0.1.89", + "version": "0.1.91", "lockfileVersion": 2, "requires": true, "packages": { "": { "name": "@scrypted/ncnn", - "version": "0.1.89", + "version": "0.1.91", "devDependencies": { "@scrypted/sdk": "file:../../sdk" } diff --git a/plugins/ncnn/package.json b/plugins/ncnn/package.json index 9031734fa..6fccc61a8 100644 --- a/plugins/ncnn/package.json +++ b/plugins/ncnn/package.json @@ -50,5 +50,5 @@ "devDependencies": { "@scrypted/sdk": "file:../../sdk" }, - "version": "0.1.89" + "version": "0.1.91" } diff --git a/plugins/ncnn/src/nc/face_recognition.py b/plugins/ncnn/src/nc/face_recognition.py index fcc8d4c29..614497d09 100644 --- a/plugins/ncnn/src/nc/face_recognition.py +++ b/plugins/ncnn/src/nc/face_recognition.py @@ -23,7 +23,7 @@ class NCNNFaceRecognition(FaceRecognizeDetection): def downloadModel(self, model: str): scrypted_yolov9 = "scrypted_yolov9" in model - ncnnmodel = "best_converted" if scrypted_yolov9 else "best" + ncnnmodel = "best_converted" if scrypted_yolov9 else model model_path = self.downloadHuggingFaceModelLocalFallback(model) binFile = os.path.join(model_path, f"{ncnnmodel}.ncnn.bin") paramFile = os.path.join(model_path, f"{ncnnmodel}.ncnn.param") diff --git a/plugins/openvino/src/predict/face_recognize.py b/plugins/openvino/src/predict/face_recognize.py index 495de4e3f..d90b25e00 100644 --- a/plugins/openvino/src/predict/face_recognize.py +++ b/plugins/openvino/src/predict/face_recognize.py @@ -35,8 +35,12 @@ class FaceRecognizeDetection(PredictPlugin): self.loop = asyncio.get_event_loop() self.minThreshold = 0.5 - self.detectModel = self.downloadModel("scrypted_yolov9t_relu_face") - self.faceModel = self.downloadModel("inception_resnet_v1") + try: + self.detectModel = self.downloadModel("scrypted_yolov9t_relu_face") + self.faceModel = self.downloadModel("inception_resnet_v1") + except Exception: + traceback.print_exc() + raise def downloadModel(self, model: str): pass