change onnx to NHWC (#1)

- change onnx to NHWC (9f86314792bc63c7d4327617e9dd76b69ed0c1b6)

RCAN_int8.onnx → RCAN_int8_NHWC.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1f80a5945e9d7bd9da2625aeec430dad3ba1123788edf36416f80ef59207c804
-size 445505
+oid sha256:9a1cad5da6396a4c812bb5b0d60c1470a1e2de1b9b4e8fe58ce4132972b164f7
+size 445692

data/data_tiling.py

@@ -18,7 +18,7 @@ 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 

eval_onnx.py

@@ -19,7 +19,7 @@ class Configs():
         # ipu test or cpu, you need to provide onnx path 
         parser.add_argument('--ipu', action='store_true',
                             help='use ipu')
-        parser.add_argument('--onnx_path', type=str, default='RCAN_int8.onnx',
+        parser.add_argument('--onnx_path', type=str, default='RCAN_int8_NHWC.onnx',
                             help='onnx path')
         parser.add_argument('--provider_config', type=str, default=None,
                             help='provider config path')

infer_onnx.py

@@ -31,7 +31,7 @@ def main(args):
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser(description='RCAN SISR')
-    parser.add_argument('--onnx_path', type=str, default='RCAN_int8.onnx',
+    parser.add_argument('--onnx_path', type=str, default='RCAN_int8_NHWC.onnx',
                     help='onnx path')
     parser.add_argument('--image_path', default='test_data/test.png',
                     help='path of your image')