First working version.

app.py

@@ -14,7 +14,7 @@ model = None
 
 def load_model():
     global model
-    model = torch.jit.load("model.pt")
+    model = torch.jit.load("v4_gated_unet-rn18-rn18_mse_bn32_5e-3_1e-5.pt")
 
 def denormalize_and_clip_as_tensor (im: Tensor) -> Tensor:
         return torch.clip(im * 0.5 + 0.5, 0.0, 1.0)
@@ -23,17 +23,43 @@ def denormalize_and_clip_as_numpy (im: Tensor) -> np.ndarray:
     im = im.squeeze(0)
     return np.ascontiguousarray(denormalize_and_clip_as_tensor(im).permute(1,2,0).detach().cpu().numpy())
 
+# pad image to a multiple of 64
+def pad_image(im, multiple=64):
+    # B,C,H,W
+    rows = im.shape[2]
+    cols = im.shape[3]
+    if rows % multiple == 0 and cols % multiple == 0:
+        return im
+    else:
+        rows_to_pad = multiple - (rows % multiple)
+        cols_to_pad = multiple - (cols % multiple)
+        return transforms.Pad(padding=(0, 0, cols_to_pad, rows_to_pad), padding_mode='reflect')(im)
+
 def undo_antialiasing(im):
     im_torch = torch.from_numpy (im).permute(2,0,1).unsqueeze(0).float() / 255.0
     im_torch = transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))(im_torch)
+
+    initial_rows = im.shape[0]
+    initial_cols = im.shape[1]
+    im_torch = pad_image(im_torch)
+    
     with torch.no_grad():
         output_torch = model(im_torch)
     output = denormalize_and_clip_as_numpy(output_torch.rgb)
-    return (output*255.99).astype(np.uint8)
+    output = (output*255.99).astype(np.uint8)
+    output = output[:initial_rows, :initial_cols, :]
+    return output
 
 load_model()
 iface = gr.Interface(fn=undo_antialiasing,
                      inputs=gr.inputs.Image(),
                      outputs=gr.outputs.Image(),
-                     examples=[['examples/Bowling.png'], ['examples/opencv.png']])
+                     examples=[
+                         ['examples/opencv.png'], 
+                         ['examples/matplotlib.png'], 
+                         ['examples/coco_beach.png'], 
+                         ['examples/plot-bowling.png']])
+                        # FIXME: add OpenCV-background
+                        # FIXME: add some arXiv test image
+
 iface.launch()

v4_gated_unet-rn18-rn18_mse_bn32_5e-3_1e-5.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b6e2d3cad8875825f30420111462a97de4f2f91c642f27aabbeec5abaad4590d
+size 118285630