From 4a8d5dbe6db1f1bfc04e63b24ac75def39578b5c Mon Sep 17 00:00:00 2001 From: Koushik Dutta Date: Sun, 8 Mar 2026 10:39:00 -0700 Subject: [PATCH] detect: fix segmentation labels --- plugins/coreml/package-lock.json | 4 ++-- plugins/coreml/package.json | 2 +- plugins/ncnn/package-lock.json | 4 ++-- plugins/ncnn/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/openvino/src/predict/segment.py | 6 +++++- 9 files changed, 17 insertions(+), 13 deletions(-) diff --git a/plugins/coreml/package-lock.json b/plugins/coreml/package-lock.json index 711fed707..b768bf61b 100644 --- a/plugins/coreml/package-lock.json +++ b/plugins/coreml/package-lock.json @@ -1,12 +1,12 @@ { "name": "@scrypted/coreml", - "version": "0.1.97", + "version": "0.1.98", "lockfileVersion": 2, "requires": true, "packages": { "": { "name": "@scrypted/coreml", - "version": "0.1.97", + "version": "0.1.98", "devDependencies": { "@scrypted/sdk": "file:../../sdk" } diff --git a/plugins/coreml/package.json b/plugins/coreml/package.json index 799062a5d..4a201c336 100644 --- a/plugins/coreml/package.json +++ b/plugins/coreml/package.json @@ -50,5 +50,5 @@ "devDependencies": { "@scrypted/sdk": "file:../../sdk" }, - "version": "0.1.97" + "version": "0.1.98" } diff --git a/plugins/ncnn/package-lock.json b/plugins/ncnn/package-lock.json index 058da39da..2645204ad 100644 --- a/plugins/ncnn/package-lock.json +++ b/plugins/ncnn/package-lock.json @@ -1,12 +1,12 @@ { "name": "@scrypted/ncnn", - "version": "0.1.92", + "version": "0.1.93", "lockfileVersion": 2, "requires": true, "packages": { "": { "name": "@scrypted/ncnn", - "version": "0.1.92", + "version": "0.1.93", "devDependencies": { "@scrypted/sdk": "file:../../sdk" } diff --git a/plugins/ncnn/package.json b/plugins/ncnn/package.json index 5dee3c7a8..eec0427b1 100644 --- a/plugins/ncnn/package.json +++ b/plugins/ncnn/package.json @@ -45,5 +45,5 @@ "devDependencies": { "@scrypted/sdk": "file:../../sdk" }, - "version": "0.1.92" + "version": "0.1.93" } diff --git a/plugins/onnx/package-lock.json b/plugins/onnx/package-lock.json index cc6aed7c6..5194a7aef 100644 --- a/plugins/onnx/package-lock.json +++ b/plugins/onnx/package-lock.json @@ -1,12 +1,12 @@ { "name": "@scrypted/onnx", - "version": "0.1.134", + "version": "0.1.135", "lockfileVersion": 2, "requires": true, "packages": { "": { "name": "@scrypted/onnx", - "version": "0.1.134", + "version": "0.1.135", "devDependencies": { "@scrypted/sdk": "file:../../sdk" } diff --git a/plugins/onnx/package.json b/plugins/onnx/package.json index d7cd0e151..8a9ceb13a 100644 --- a/plugins/onnx/package.json +++ b/plugins/onnx/package.json @@ -50,5 +50,5 @@ "devDependencies": { "@scrypted/sdk": "file:../../sdk" }, - "version": "0.1.134" + "version": "0.1.135" } diff --git a/plugins/openvino/package-lock.json b/plugins/openvino/package-lock.json index c4302e64a..1494a152a 100644 --- a/plugins/openvino/package-lock.json +++ b/plugins/openvino/package-lock.json @@ -1,12 +1,12 @@ { "name": "@scrypted/openvino", - "version": "0.1.198", + "version": "0.1.199", "lockfileVersion": 2, "requires": true, "packages": { "": { "name": "@scrypted/openvino", - "version": "0.1.198", + "version": "0.1.199", "devDependencies": { "@scrypted/sdk": "file:../../sdk" } diff --git a/plugins/openvino/package.json b/plugins/openvino/package.json index 6e195ab20..4b551a648 100644 --- a/plugins/openvino/package.json +++ b/plugins/openvino/package.json @@ -50,5 +50,5 @@ "devDependencies": { "@scrypted/sdk": "file:../../sdk" }, - "version": "0.1.198" + "version": "0.1.199" } diff --git a/plugins/openvino/src/predict/segment.py b/plugins/openvino/src/predict/segment.py index 6b20587a6..8c3a415b4 100644 --- a/plugins/openvino/src/predict/segment.py +++ b/plugins/openvino/src/predict/segment.py @@ -21,7 +21,11 @@ class Segmentation(PredictPlugin): self.inputwidth = 320 self.inputheight = 320 self.loop = asyncio.get_event_loop() - self.labels = coco.COCO_LABELS + self.labels = { + 0: "person", + 1: "vehicle", + 2: "animal", + } try: self.model = self.loadModel('scrypted_yolov9t_seg_relu')