Update README.md
#2
by
wensongc
- opened
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='
|
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
|
99 |
```
|
100 |
|
101 |
- Test accuracy of the quantized model
|
102 |
```python
|
103 |
-
python test_onnx.py --onnx_path
|
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 |
|