From 684961fa4b348673ee077b700bf41904699d94df Mon Sep 17 00:00:00 2001 From: Koushik Dutta Date: Fri, 24 May 2024 22:43:11 -0700 Subject: [PATCH] openvino: types --- plugins/openvino/src/common/yolo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/plugins/openvino/src/common/yolo.py b/plugins/openvino/src/common/yolo.py index 05a6d8287..23f0c2a66 100644 --- a/plugins/openvino/src/common/yolo.py +++ b/plugins/openvino/src/common/yolo.py @@ -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]