change to NHWC

data/data_tiling.py

@@ -24,7 +24,10 @@ def tiling_inference(session, lr, overlapping, patch_size):
             w_idx = w_idx if w_idx + patch_size[1] <= w else w - patch_size[1]
 
             tilling_lr = lr[..., h_idx: h_idx+patch_size[0], w_idx: w_idx+patch_size[1]]
-            sr_tiling = session.run(None, {session.get_inputs()[0].name: tilling_lr})[0]
+            sr_tiling = session.run(None, {session.get_inputs()[0].name: tilling_lr.transpose(0,2,3,1)})[0].transpose(0,3,1,2)
+
+            left, right, top, bottom = 0, patch_size[1], 0, patch_size[0]
+            left += overlapping//2 
 
             left, right, top, bottom = 0, patch_size[1], 0, patch_size[0]
             left += overlapping//2