From e1f558a8aab38803a493489c4487b1eccff84872 Mon Sep 17 00:00:00 2001
From: David Chatel <dchatel@live.fr>
Date: Sun, 10 Mar 2024 13:35:22 +0100
Subject: [PATCH] fixed padding bug

---
 utils.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/utils.py b/utils.py
index a3e4bd9..6dc996f 100644
--- a/utils.py
+++ b/utils.py
@@ -43,9 +43,9 @@ def get_submatrix_with_padding(img, a, b, c, d):
     pb = -min(img.shape[0] - d, 0)
     a, b, c, d = max(a, 0), max(b, 0), min(c, img.shape[1]), min(d, img.shape[0])
 
-    submatrix = img[b:d, a:c]
+    submatrix = img[b:d, a:c].permute(2,0,1)
     pad = tv.transforms.Pad((pl, pt, pr, pb))
-    submatrix = pad(submatrix)
+    submatrix = pad(submatrix).permute(1,2,0)
     
     return submatrix