From 95c8576aaf74a52f15afd769a5d08031d39b06bb Mon Sep 17 00:00:00 2001 From: kirilllzaitsev Date: Mon, 31 Jul 2023 13:09:35 +0200 Subject: [PATCH] fix mask shape (N, W, H) -> (N, H, W) --- supervision/dataset/formats/pascal_voc.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/supervision/dataset/formats/pascal_voc.py b/supervision/dataset/formats/pascal_voc.py index e915159c..93e90927 100644 --- a/supervision/dataset/formats/pascal_voc.py +++ b/supervision/dataset/formats/pascal_voc.py @@ -161,7 +161,7 @@ def load_pascal_voc_annotations( tree = parse(annotation_path) root = tree.getroot() - resolution_wh = (image.shape[0], image.shape[1]) + resolution_wh = (image.shape[1], image.shape[0]) annotation, classes = detections_from_xml_obj( root, classes, resolution_wh, force_masks )