michael/scrypted
1
0
Fork 0
mirror of https://github.com/koush/scrypted.git synced 2026-08-13 00:20:45 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

openvino: types

Browse Source
This commit is contained in:
Koushik Dutta
2024-05-24 22:43:11 -07:00
parent 4f60b7e379
commit 684961fa4b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
plugins/openvino/src/common/yolo.py
View File

@@ -21,7 +21,7 @@ def parse_yolo_nas(predictions):
return objs
def parse_yolov9(results, threshold = defaultThreshold, scale = None, confidence_scale = None):
objs = []
objs: list[Prediction] = []
keep = np.argwhere(results[4:] > threshold)
for indices in keep:
class_id = indices[0]