From f4b91d0175eb3c08ace30cb58bdcb1dfbfe7795c Mon Sep 17 00:00:00 2001 From: SkalskiP Date: Mon, 8 May 2023 19:46:13 +0200 Subject: [PATCH] Fix load_yolo_annotations --- supervision/dataset/formats/yolo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/supervision/dataset/formats/yolo.py b/supervision/dataset/formats/yolo.py index 0002d6d1..57c850ef 100644 --- a/supervision/dataset/formats/yolo.py +++ b/supervision/dataset/formats/yolo.py @@ -135,7 +135,7 @@ def load_yolo_annotations( for image_path, annotation_path in path_pairs: image = cv2.imread(str(image_path)) lines = read_txt_file(str(annotation_path)) - w, h, _ = image.shape + h, w, _ = image.shape resolution_wh = (w, h) with_masks = _with_mask(lines=lines)