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]