mirror of
https://github.com/koush/scrypted.git
synced 2026-03-16 15:12:06 +00:00
detect: fix segmentation labels
This commit is contained in:
4
plugins/coreml/package-lock.json
generated
4
plugins/coreml/package-lock.json
generated
@@ -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"
|
||||
}
|
||||
|
||||
@@ -50,5 +50,5 @@
|
||||
"devDependencies": {
|
||||
"@scrypted/sdk": "file:../../sdk"
|
||||
},
|
||||
"version": "0.1.97"
|
||||
"version": "0.1.98"
|
||||
}
|
||||
|
||||
4
plugins/ncnn/package-lock.json
generated
4
plugins/ncnn/package-lock.json
generated
@@ -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"
|
||||
}
|
||||
|
||||
@@ -45,5 +45,5 @@
|
||||
"devDependencies": {
|
||||
"@scrypted/sdk": "file:../../sdk"
|
||||
},
|
||||
"version": "0.1.92"
|
||||
"version": "0.1.93"
|
||||
}
|
||||
|
||||
4
plugins/onnx/package-lock.json
generated
4
plugins/onnx/package-lock.json
generated
@@ -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"
|
||||
}
|
||||
|
||||
@@ -50,5 +50,5 @@
|
||||
"devDependencies": {
|
||||
"@scrypted/sdk": "file:../../sdk"
|
||||
},
|
||||
"version": "0.1.134"
|
||||
"version": "0.1.135"
|
||||
}
|
||||
|
||||
4
plugins/openvino/package-lock.json
generated
4
plugins/openvino/package-lock.json
generated
@@ -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"
|
||||
}
|
||||
|
||||
@@ -50,5 +50,5 @@
|
||||
"devDependencies": {
|
||||
"@scrypted/sdk": "file:../../sdk"
|
||||
},
|
||||
"version": "0.1.198"
|
||||
"version": "0.1.199"
|
||||
}
|
||||
|
||||
@@ -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')
|
||||
|
||||
Reference in New Issue
Block a user