zhengrongzhang wensongc commited on
Commit
1b18724
1 Parent(s): e66aed5

Update README.md (#2)

Browse files

- Update README.md (9c96efc04744a5b5ad01feb4d38e85e5e8360732)


Co-authored-by: Wensong Chan <wensongc@users.noreply.huggingface.co>

Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -63,7 +63,7 @@ You can use the raw model for image segmentation. See the [model hub](https://hu
63
  - Code snippet from [`infer_onnx.py`](infer_onnx.py) on how to use
64
  ```python
65
  parser = argparse.ArgumentParser(description='SemanticFPN model')
66
- parser.add_argument('--onnx_path', type=str, default='FPN_int.onnx')
67
  parser.add_argument('--save_path', type=str, default='./data/demo_results/senmatic_results.png')
68
  parser.add_argument('--input_path', type=str, default='data/cityscapes/cityscapes/leftImg8bit/test/bonn/bonn_000000_000019_leftImg8bit.png')
69
  parser.add_argument('--ipu', action='store_true',
@@ -95,12 +95,12 @@ You can use the raw model for image segmentation. See the [model hub](https://hu
95
 
96
  - Run inference for a single image
97
  ```python
98
- python infer_onnx.py --onnx_path FPN_int.onnx --input_path /Path/To/Your/Image --ipu --provider_config Path/To/vaip_config.json
99
  ```
100
 
101
  - Test accuracy of the quantized model
102
  ```python
103
- python test_onnx.py --onnx_path FPN_int.onnx --dataset citys --test-folder ./data/cityscapes --crop-size 256 --ipu --provider_config Path/To/vaip_config.json
104
  ```
105
  ### Performance
106
 
 
63
  - Code snippet from [`infer_onnx.py`](infer_onnx.py) on how to use
64
  ```python
65
  parser = argparse.ArgumentParser(description='SemanticFPN model')
66
+ parser.add_argument('--onnx_path', type=str, default='FPN_int_NHWC.onnx')
67
  parser.add_argument('--save_path', type=str, default='./data/demo_results/senmatic_results.png')
68
  parser.add_argument('--input_path', type=str, default='data/cityscapes/cityscapes/leftImg8bit/test/bonn/bonn_000000_000019_leftImg8bit.png')
69
  parser.add_argument('--ipu', action='store_true',
 
95
 
96
  - Run inference for a single image
97
  ```python
98
+ python infer_onnx.py --onnx_path FPN_int_NHWC.onnx --input_path /Path/To/Your/Image --ipu --provider_config Path/To/vaip_config.json
99
  ```
100
 
101
  - Test accuracy of the quantized model
102
  ```python
103
+ python test_onnx.py --onnx_path FPN_int_NHWC.onnx --dataset citys --test-folder ./data/cityscapes --crop-size 256 --ipu --provider_config Path/To/vaip_config.json
104
  ```
105
  ### Performance
106