README.md 1021 字节
Newer Older
J
Jason 已提交
1 2
# onnx2fluid
[![License](https://img.shields.io/badge/license-Apache%202-blue.svg)](LICENSE)
M
Macrobull 已提交
3

J
Jason 已提交
4 5 6 7 8 9 10 11 12

onnx2fluid支持将onnx模型转换为PaddlePaddle模型,并用于预测。

## 环境安装

工具开发过程中,我们在如下环境配置中测试模型转换,建议使用[anaconda](https://docs.anaconda.com/anaconda/install)

> python2 & python3

J
Jason 已提交
13
> onnx == 1.4.0
J
Jason 已提交
14 15 16 17

> paddlepaddle == 1.3.0

``` shell
J
Jason 已提交
18
# 安装onnx
J
Jason 已提交
19
# 也可参考https://github.com/onnx/onnx
J
Jason 已提交
20
conda install -c conda-forge onnx
J
Jason 已提交
21
```
J
Jason 已提交
22 23

## 使用说明
J
Jason 已提交
24
         
J
Jason 已提交
25 26 27
```shell
python -m onnx2fluid -e -o /path/to/export/model /path/of/onnx/model
```
J
Jason 已提交
28
**VGG19转换**
J
Jason 已提交
29 30 31 32 33 34 35 36
```shell
# 下载并解压onnx模型vgg19
wget https://s3.amazonaws.com/download.onnx/models/opset_9/vgg19.tar.gz
tar xzvf vgg19.tar.gz

# 转换为PaddlePaddle模型
python -m onnx2fluid -e -o paddle_model vgg19/model.onnx
```
J
Jason 已提交
37
转换后的PaddlePaddle模型加载可参考文档[加载预测模型](http://www.paddlepaddle.org/documentation/docs/zh/1.3/api_guides/low_level/inference.html#id4)