detect: fix segmentation labels

This commit is contained in:
Koushik Dutta
2026-03-08 10:39:00 -07:00
parent c77bbec748
commit 4a8d5dbe6d
9 changed files with 17 additions and 13 deletions

View File

@@ -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"
}

View File

@@ -50,5 +50,5 @@
"devDependencies": {
"@scrypted/sdk": "file:../../sdk"
},
"version": "0.1.97"
"version": "0.1.98"
}

View File

@@ -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"
}

View File

@@ -45,5 +45,5 @@
"devDependencies": {
"@scrypted/sdk": "file:../../sdk"
},
"version": "0.1.92"
"version": "0.1.93"
}

View File

@@ -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"
}

View File

@@ -50,5 +50,5 @@
"devDependencies": {
"@scrypted/sdk": "file:../../sdk"
},
"version": "0.1.134"
"version": "0.1.135"
}

View File

@@ -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"
}

View File

@@ -50,5 +50,5 @@
"devDependencies": {
"@scrypted/sdk": "file:../../sdk"
},
"version": "0.1.198"
"version": "0.1.199"
}

View File

@@ -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')