1

app.py

@@ -35,29 +35,37 @@ def predict1(img):
     b, c, h, w = in_img.size()
     # pad image such that the resolution is a multiple of 32
     w_pad = (math.ceil(w / 32) * 32 - w) // 2
+    w_odd_pad = w_pad
     h_pad = (math.ceil(h / 32) * 32 - h) // 2
-    in_img = img_pad(in_img, w_r=w_pad, h_r=h_pad)
+    h_odd_pad = h_pad
+
+    if w % 2 == 1:
+        w_odd_pad += 1
+    if h % 2 == 1:
+        h_odd_pad += 1
+
+    in_img = img_pad(in_img, w_pad=w_pad, h_pad=h_pad, w_odd_pad=w_odd_pad, h_odd_pad=h_odd_pad)
     with torch.no_grad():
         out_1, out_2, out_3 = model1(in_img)
         if h_pad != 0:
-            out_1 = out_1[:, :, h_pad:-h_pad, :]
+            out_1 = out_1[:, :, h_pad:-h_odd_pad, :]
         if w_pad != 0:
-            out_1 = out_1[:, :, :, w_pad:-w_pad]
+            out_1 = out_1[:, :, :, w_pad:-w_odd_pad]
     out_1 = out_1.squeeze(0)
     out_1 = PIL.Image.fromarray(torch.clamp(out_1 * 255, min=0, max=255
     ).byte().permute(1, 2, 0).cpu().numpy())
 
     return out_1
 
-def img_pad(x, h_r=0, w_r=0):
+def img_pad(x,  w_pad, h_pad, w_odd_pad, h_odd_pad):
     '''
     Here the padding values are determined by the average r,g,b values across the training set
     in FHDMi dataset. For the evaluation on the UHDM, you can also try the commented lines where
     the mean values are calculated from UHDM training set, yielding similar performance.
     '''
-    x1 = F.pad(x[:, 0:1, ...], (w_r, w_r, h_r, h_r), value=0.3827)
-    x2 = F.pad(x[:, 1:2, ...], (w_r, w_r, h_r, h_r), value=0.4141)
-    x3 = F.pad(x[:, 2:3, ...], (w_r, w_r, h_r, h_r), value=0.3912)
+    x1 = F.pad(x[:, 0:1, ...], (w_pad, w_odd_pad, h_pad, h_odd_pad), value=0.3827)
+    x2 = F.pad(x[:, 1:2, ...], (w_pad, w_odd_pad, h_pad, h_odd_pad), value=0.4141)
+    x3 = F.pad(x[:, 2:3, ...], (w_pad, w_odd_pad, h_pad, h_odd_pad), value=0.3912)
 
     y = torch.cat([x1, x2, x3], dim=1)
 
@@ -87,7 +95,7 @@ iface1 = gr.Interface(fn=predict1,
                             '003.jpg',
                             '004.jpg',
                             '005.jpg'],
-                    title = title, 
+                    title = title,
                     description = description,
                     article = article
                     )