Views
No views yet
pip install -r requirements.txt 1└── dataset
2 └── benchmark
3 ├── Set5
4 ├── HR
5 | ├── baby.png
6 | ├── ...
7 └── LR_bicubic
8 └──X2
9 ├──babyx2.png
10 ├── ...
11 ├── Set14
12 ├── ... infer_onnx.py on how to use1 parser = argparse.ArgumentParser(description='RCAN SISR')
2 parser.add_argument('--onnx_path', type=str, default='RCAN_int8_NHWC.onnx',
3 help='onnx path')
4 parser.add_argument('--image_path', default='test_data/test.png',
5 help='path of your image')
6 parser.add_argument('--output_path', default='test_data/sr.png',
7 help='path of your image')
8 parser.add_argument('--ipu', action='store_true',
9 help='use ipu')
10 parser.add_argument('--provider_config', type=str, default=None,
11 help='provider config path')
12 args = parser.parse_args()
13
14 if args.ipu:
15 providers = ["VitisAIExecutionProvider"]
16 provider_options = [{"config_file": args.provider_config}]
17 else:
18 providers = ['CUDAExecutionProvider', 'CPUExecutionProvider']
19 provider_options = None
20 onnx_file_name = args.onnx_path
21 image_path = args.image_path
22 output_path = args.output_path
23
24 ort_session = onnxruntime.InferenceSession(onnx_file_name, providers=providers, provider_options=provider_options)
25 lr = cv2.imread(image_path)[np.newaxis,:,:,:].transpose((0,3,1,2)).astype(np.float32)
26 sr = tiling_inference(ort_session, lr, 8, (56, 56))
27 sr = np.clip(sr, 0, 255)
28 sr = sr.squeeze().transpose((1,2,0)).astype(np.uint8)
29 sr = cv2.imwrite(output_path, sr)python infer_onnx.py --onnx_path RCAN_int8_NHWC.onnx --image_path /Path/To/Your/Image --ipu --provider_config Path/To/vaip_config.jsonpython eval_onnx.py --onnx_path RCAN_int8_NHWC.onnx --data_test Set5 --ipu --provider_config Path/To/vaip_config.json | Method | Scale | Flops | Set5 |
|---|---|---|---|
| RCAN-S (float) | X2 | 24.5G | 37.531 / 0.958 |
| RCAN-S (INT8) | X2 | 24.5G | 37.150 / 0.955 |
1@inproceedings{zhang2018image,
2 title={Image super-resolution using very deep residual channel attention networks},
3 author={Zhang, Yulun and Li, Kunpeng and Li, Kai and Wang, Lichen and Zhong, Bineng and Fu, Yun},
4 booktitle={Proceedings of the European conference on computer vision (ECCV)},
5 pages={286--301},
6 year={2018}
7}