Fix load_yolo_annotations

This commit is contained in:
SkalskiP committed 2023-05-08 19:46:13 +02:00
1 parent 45fdb1e805
commit f4b91d0175
1 file changed
+1 -1
+1 -1
View File
@@ -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)