fix mask shape (N, W, H) -> (N, H, W)
This commit is contained in:
1 parent
12b3def032
commit
95c8576aaf
1 file changed
+1
-1
@@ -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
|
||||
)
|
||||
|
||||
Reference in new issue
Block a user