fix mask shape (N, W, H) -> (N, H, W)

This commit is contained in:
kirilllzaitsev committed 2023-07-31 13:09:35 +02:00
1 parent 12b3def032
commit 95c8576aaf
1 file changed
+1 -1
+1 -1
View File
@@ -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
)