Skip to content

Commit 0d27045

Browse files
committed
predict: switch everything to yolov9t/c
1 parent e740a69 commit 0d27045

File tree

12 files changed

+33
-25
lines changed

12 files changed

+33
-25
lines changed

plugins/coreml/package-lock.json

+2-2
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

plugins/coreml/package.json

+1-1
Original file line numberDiff line numberDiff line change
@@ -42,5 +42,5 @@
4242
"devDependencies": {
4343
"@scrypted/sdk": "file:../../sdk"
4444
},
45-
"version": "0.1.64"
45+
"version": "0.1.65"
4646
}

plugins/coreml/src/coreml/__init__.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,8 @@
3131
"scrypted_yolo_nas_s_320",
3232
"scrypted_yolov9e_320",
3333
"scrypted_yolov9c_320",
34+
"scrypted_yolov9s_320",
35+
"scrypted_yolov9t_320",
3436
"scrypted_yolov6n_320",
3537
"scrypted_yolov6s_320",
3638
"scrypted_yolov8n_320",
@@ -81,7 +83,7 @@ def __init__(self, nativeId: str | None = None):
8183
self.scrypted_yolo_nas = "scrypted_yolo_nas" in model
8284
self.scrypted_yolo = "scrypted_yolo" in model
8385
self.scrypted_model = "scrypted" in model
84-
model_version = "v7"
86+
model_version = "v8"
8587
mlmodel = "model" if self.scrypted_yolo else model
8688

8789
print(f"model: {model}")

plugins/onnx/.vscode/settings.json

+8-6
Original file line numberDiff line numberDiff line change
@@ -1,18 +1,20 @@
11

22
{
33
// docker installation
4-
// "scrypted.debugHost": "koushik-ubuntuvm",
5-
// "scrypted.serverRoot": "/server",
6-
// "scrypted.debugHost": "koushik-ubuntuvm",
7-
// "scrypted.serverRoot": "/home/koush/.scrypted",
4+
"scrypted.debugHost": "koushik-ubuntuvm",
5+
"scrypted.serverRoot": "/server",
6+
7+
// lxc
8+
// "scrypted.debugHost": "scrypted-server",
9+
// "scrypted.serverRoot": "/root/.scrypted",
810

911
// pi local installation
1012
// "scrypted.debugHost": "192.168.2.119",
1113
// "scrypted.serverRoot": "/home/pi/.scrypted",
1214

1315
// local checkout
14-
"scrypted.debugHost": "127.0.0.1",
15-
"scrypted.serverRoot": "/Users/koush/.scrypted",
16+
// "scrypted.debugHost": "127.0.0.1",
17+
// "scrypted.serverRoot": "/Users/koush/.scrypted",
1618
// "scrypted.debugHost": "koushik-winvm",
1719
// "scrypted.serverRoot": "C:\\Users\\koush\\.scrypted",
1820

plugins/onnx/package-lock.json

+2-2
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

plugins/onnx/package.json

+1-1
Original file line numberDiff line numberDiff line change
@@ -42,5 +42,5 @@
4242
"devDependencies": {
4343
"@scrypted/sdk": "file:../../sdk"
4444
},
45-
"version": "0.1.102"
45+
"version": "0.1.103"
4646
}

plugins/onnx/src/ort/__init__.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,8 @@
3535
"scrypted_yolov6n_320",
3636
"scrypted_yolov6s_320",
3737
"scrypted_yolov9c_320",
38+
"scrypted_yolov9s_320",
39+
"scrypted_yolov9t_320",
3840
"scrypted_yolov8n_320",
3941
]
4042

@@ -55,7 +57,7 @@ def __init__(self, nativeId: str | None = None):
5557
if model == "Default" or model not in availableModels:
5658
if model != "Default":
5759
self.storage.setItem("model", "Default")
58-
model = "scrypted_yolov10m_320"
60+
model = "scrypted_yolov9c_320"
5961
self.yolo = "yolo" in model
6062
self.scrypted_yolov10 = "scrypted_yolov10" in model
6163
self.scrypted_yolo_nas = "scrypted_yolo_nas" in model
@@ -66,7 +68,7 @@ def __init__(self, nativeId: str | None = None):
6668

6769
onnxmodel = model if self.scrypted_yolo_nas else "best" if self.scrypted_model else model
6870

69-
model_version = "v2"
71+
model_version = "v3"
7072
onnxfile = self.downloadFile(
7173
f"https://github.com/koush/onnx-models/raw/main/{model}/{onnxmodel}.onnx",
7274
f"{model_version}/{model}/{onnxmodel}.onnx",

plugins/openvino/.vscode/settings.json

+4-4
Original file line numberDiff line numberDiff line change
@@ -5,16 +5,16 @@
55
// "scrypted.serverRoot": "/server",
66

77
// proxmox installation
8-
// "scrypted.debugHost": "scrypted-server",
9-
// "scrypted.serverRoot": "/root/.scrypted",
8+
"scrypted.debugHost": "scrypted-server",
9+
"scrypted.serverRoot": "/root/.scrypted",
1010

1111
// pi local installation
1212
// "scrypted.debugHost": "192.168.2.119",
1313
// "scrypted.serverRoot": "/home/pi/.scrypted",
1414

1515
// local checkout
16-
"scrypted.debugHost": "127.0.0.1",
17-
"scrypted.serverRoot": "/Users/koush/.scrypted",
16+
// "scrypted.debugHost": "127.0.0.1",
17+
// "scrypted.serverRoot": "/Users/koush/.scrypted",
1818
// "scrypted.debugHost": "koushik-winvm",
1919
// "scrypted.serverRoot": "C:\\Users\\koush\\.scrypted",
2020

plugins/openvino/package-lock.json

+2-2
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

plugins/openvino/package.json

+1-1
Original file line numberDiff line numberDiff line change
@@ -42,5 +42,5 @@
4242
"devDependencies": {
4343
"@scrypted/sdk": "file:../../sdk"
4444
},
45-
"version": "0.1.103"
45+
"version": "0.1.104"
4646
}

plugins/openvino/src/ov/__init__.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,8 @@
3737
"scrypted_yolov6n_320",
3838
"scrypted_yolov6s_320",
3939
"scrypted_yolov9c_320",
40+
"scrypted_yolov9s_320",
41+
"scrypted_yolov9t_320",
4042
"scrypted_yolov8n_320",
4143
"ssd_mobilenet_v1_coco",
4244
"ssdlite_mobilenet_v2",
@@ -135,7 +137,7 @@ def __init__(self, nativeId: str | None = None):
135137
if model == "Default" or model not in availableModels:
136138
if model != "Default":
137139
self.storage.setItem("model", "Default")
138-
model = "scrypted_yolov10n_320"
140+
model = "scrypted_yolov9t_320"
139141
self.yolo = "yolo" in model
140142
self.scrypted_yolov10 = "scrypted_yolov10" in model
141143
self.scrypted_yolo_nas = "scrypted_yolo_nas" in model

plugins/openvino/src/predict/face_recognize.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def __init__(self, nativeId: str | None = None):
3535
self.loop = asyncio.get_event_loop()
3636
self.minThreshold = 0.5
3737

38-
self.detectModel = self.downloadModel("scrypted_yolov10n_face_320")
38+
self.detectModel = self.downloadModel("scrypted_yolov9t_face_320")
3939
self.faceModel = self.downloadModel("inception_resnet_v1")
4040

4141
def downloadModel(self, model: str):
@@ -53,7 +53,7 @@ def get_input_format(self) -> str:
5353

5454
async def detect_once(self, input: Image.Image, settings: Any, src_size, cvss):
5555
results = await self.predictDetectModel(input)
56-
objs = yolo.parse_yolov10(results)
56+
objs = yolo.parse_yolov9(results)
5757
ret = self.create_detection_result(objs, src_size, cvss)
5858
return ret
5959

0 commit comments

Comments
 (0)