readme.md 661 Bytes
Newer Older
wangkaixiong's avatar
wangkaixiong committed
1
2
3
4
5
6
7
8
9
10
## 方法一

使用 `onnxsim` 转换;

安装包的步骤
```bash
pip install onnx-simplifier
```

```python
wangkx1's avatar
wangkx1 committed
11
# 简化模型
wangkaixiong's avatar
wangkaixiong committed
12
python -m onnxsim ../resnet50.onnx ../resnet50_1x3x224x224.onnx --test-input-shape input:1,3,224,224 
wangkx1's avatar
wangkx1 committed
13
14
# 动态shape变成静态shape
python -m onnxsim ./${model_name}_simply.onnx ./${model_name}_simply_1x3x32x900.onnx --overwrite-input-shape x:1,3,32,900
wangkaixiong's avatar
wangkaixiong committed
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

# 查看详细帮助内容
python -m onnxsim -h

```

## 方法二

使用 `paddle2onnx` 去进行转换。

```python
pip3 install paddle2onnx

python -m paddle2onnx.optimize --input_model model.onnx \
--output_model new_model.onnx \
--input_shape_dict "{'x':[1,3,224,224]}"
wangkx1's avatar
wangkx1 committed
31
```