openvino: types

koush · May 25, 2024 · 684961f · 684961f
1 parent 4f60b7e
commit 684961f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]