Cosmetic changes.

kaanaksit · Nov 30, 2023 · b4a86b5 · b4a86b5
1 parent f4e8deb
commit b4a86b5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/odak/learn/tools/matrix.py b/odak/learn/tools/matrix.py
@@ -119,7 +119,7 @@ def crop_center(field, size = None):
         cropped_padded = field[:, :, cx-hx:cx+hx, cy-hy:cy+hy]
     cropped = cropped_padded
     if permute_flag == True:
-        field = field.permute(0, 3, 1, 2)
+        field = field.permute(0, 2, 3, 1)
     if len(orig_resolution) == 2:
         cropped = cropped_padded.squeeze(0).squeeze(0)
     if len(orig_resolution) == 3: