zhengrongzhang wensongc commited on
Commit
06ecfa4
1 Parent(s): 30c4d88

Update README.md (#2)

Browse files

- Update README.md (00e855d3f2f7193f80d7f96b13d5440cce4d026b)


Co-authored-by: Wensong Chan <[email protected]>

Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -66,7 +66,7 @@ You can use the raw model for super resolution. See the [model hub](https://hugg
66
  - Code snippet from [`infer_onnx.py`](infer_onnx.py) on how to use
67
  ```python
68
  parser = argparse.ArgumentParser(description='RCAN SISR')
69
- parser.add_argument('--onnx_path', type=str, default='RCAN_int8.onnx',
70
  help='onnx path')
71
  parser.add_argument('--image_path', default='test_data/test.png',
72
  help='path of your image')
@@ -98,12 +98,12 @@ You can use the raw model for super resolution. See the [model hub](https://hugg
98
 
99
  - Run inference for a single image
100
  ```python
101
- python infer_onnx.py --onnx_path RCAN_int8.onnx --image_path /Path/To/Your/Image --ipu --provider_config Path/To/vaip_config.json
102
  ```
103
 
104
  - Test accuracy of the quantized model
105
  ```python
106
- python eval_onnx.py --onnx_path RCAN_int8.onnx --data_test Set5 --ipu --provider_config Path/To/vaip_config.json
107
  ```
108
  ### Performance
109
 
 
66
  - Code snippet from [`infer_onnx.py`](infer_onnx.py) on how to use
67
  ```python
68
  parser = argparse.ArgumentParser(description='RCAN SISR')
69
+ parser.add_argument('--onnx_path', type=str, default='RCAN_int8_NHWC.onnx',
70
  help='onnx path')
71
  parser.add_argument('--image_path', default='test_data/test.png',
72
  help='path of your image')
 
98
 
99
  - Run inference for a single image
100
  ```python
101
+ python infer_onnx.py --onnx_path RCAN_int8_NHWC.onnx --image_path /Path/To/Your/Image --ipu --provider_config Path/To/vaip_config.json
102
  ```
103
 
104
  - Test accuracy of the quantized model
105
  ```python
106
+ python eval_onnx.py --onnx_path RCAN_int8_NHWC.onnx --data_test Set5 --ipu --provider_config Path/To/vaip_config.json
107
  ```
108
  ### Performance
109