未验证 提交 2cb48107 编写于 作者: D daquexian 提交者: GitHub

upgrade onnxsim to v0.4.1 (#6473)

Signed-off-by: Ndaquexian <daquexian566@gmail.com>
上级 714b339b
......@@ -194,18 +194,13 @@ paddle2onnx --model_dir output_inference/picodet_s_320_coco_lcnet/ \
- 简化ONNX模型: 使用`onnx-simplifier`库来简化ONNX模型。
- 安装 onnx-simplifier >= 0.3.6:
- 安装 onnxsim >= 0.4.1:
```shell
pip install onnx-simplifier
pip install onnxsim
```
- 简化ONNX模型:
```shell
python -m onnxsim picodet_s_320_coco.onnx picodet_s_processed.onnx
```
如果模型包含所有后处理,简化模型时需要指定`dynamic-input-shape`
```shell
python -m onnxsim picodet_s_320_coco.onnx picodet_s_processed.onnx --dynamic-input-shape --input-shape image:1,3,320,320
onnxsim picodet_s_320_coco.onnx picodet_s_processed.onnx
```
</details>
......
......@@ -189,18 +189,13 @@ paddle2onnx --model_dir output_inference/picodet_s_320_coco_lcnet/ \
- Simplify ONNX model: use onnx-simplifier to simplify onnx model.
- Install onnx-simplifier >= 0.3.6:
- Install onnxsim >= 0.4.1:
```shell
pip install onnx-simplifier
pip install onnxsim
```
- simplify onnx model:
```shell
python -m onnxsim picodet_s_320_coco.onnx picodet_s_processed.onnx
```
If the model includes postprocessing, specify `dynamic-input-shape` when simplifying the model:
```shell
python -m onnxsim picodet_s_320_coco.onnx picodet_s_processed.onnx --dynamic-input-shape --input-shape image:1,3,320,320
onnxsim picodet_s_320_coco.onnx picodet_s_processed.onnx
```
</details>
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册