# Paddle2ONNX

**Repository Path**: xyqgcs/Paddle2ONNX

## Basic Information

- **Project Name**: Paddle2ONNX
- **Description**: paddle2onnx支持将PaddlePaddle模型格式转化到ONNX模型格式
- **Primary Language**: Unknown
- **License**: Apache-2.0
- **Default Branch**: develop
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 8
- **Created**: 2022-04-23
- **Last Updated**: 2022-04-23

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# Paddle2ONNX

[简体中文](README_zh.md) | English

## Introduction

Paddle2ONNX enables users to convert models from PaddlePaddle to ONNX.

- Supported model format. Paddle2ONNX supports both dynamic and static computational graph of PaddlePaddle. For static computational graph, Paddle2ONNX converts PaddlePaddle models saved by API [save_inference_model](https://www.paddlepaddle.org.cn/documentation/docs/zh/develop/api/paddle/static/save_inference_model_cn.html#save-inference-model), for example [IPthon example](examples/tutorial.ipynb). For dynamic computational graph, it is now under experiment and more details will be released after the release of PaddlePaddle 2.0.
- Supported operators. Paddle2ONNX can stably export models to ONNX Opset 9~11, and partialy support lower version opset. More details please refer to [Operator list](docs/en/op_list.md).
- Supported models. You can find officially verified models by Paddle2ONNX in [model zoo](docs/en/model_zoo.md).

## AIStudio Tutorials

- [Export and inference ONNX model in PaddlePaddle 2.0](https://aistudio.baidu.com/aistudio/projectdetail/1461212)
- [How to deploy PP-OCR model using ONNX RunTime](https://aistudio.baidu.com/aistudio/projectdetail/1479970)

## Environment Dependencies

### Configuration
     python >= 2.7  
     static computational graph: paddlepaddle >= 1.8.0
     dynamic computational graph: paddlepaddle >= 2.0.0
     onnx == 1.7.0 | Optional
## Installation

### Via Pip
    
     pip install paddle2onnx
    

### From Source

     git clone https://github.com/PaddlePaddle/Paddle2ONNX.git
     cd Paddle2ONNX
     python setup.py install

## Usage
### Static Computational Graph
#### Via Command Line Tool
Uncombined PaddlePaddle model(parameters saved in different files)

    paddle2onnx --model_dir paddle_model  --save_file onnx_file --opset_version 10 --enable_onnx_checker True

Combined PaddlePaddle model(parameters saved in one binary file)

    paddle2onnx --model_dir paddle_model  --model_filename model_filename --params_filename params_filename --save_file onnx_file --opset_version 10 --enable_onnx_checker True

#### Parameters
| Parameters | Description |
|----------|--------------|
|--model_dir | The directory path of the paddlepaddle model saved by `paddle.fluid.io.save_inference_model`|
|--model_filename |**[Optional]** The model file name under the directory designated by`--model_dir`. Only needed when all the model parameters saved in one binary file. Default value None|
|--params_filename |**[Optonal]** the parameter file name under the directory designated by`--model_dir`. Only needed when all the model parameters saved in one binary file. Default value None|
|--save_file | the directory path for the exported ONNX model|
|--opset_version | **[Optional]** To configure the ONNX Opset version. Opset 9-11 are stably supported. Default value is 9.|
|--enable_onnx_checker| **[Optional]**  To check the validity of the exported ONNX model. It is suggested to turn on the switch. If set to True, onnx>=1.7.0 is required. Default value is False|
|--version |**[Optional]** check the version of paddle2onnx |

- Two types of PaddlePaddle models
   - Combined model, parameters saved in one binary file. --model_filename and --params_filename represents the file name and parameter name under the directory designated by --model_dir. --model_filename and --params_filename are valid only with parameter --model_dir.
   - Uncombined model, parameters saved in different files. Only --model_dir is needed，which contains '\_\_model\_\_' file and the seperated parameter files.


#### IPython tutorials

- [Convert to ONNX from static computational graph](examples/tutorial.ipynb)

### Dynamic Computational Graph

```
import paddle
from paddle import nn
from paddle.static import InputSpec
import paddle2onnx as p2o

class LinearNet(nn.Layer):
    def __init__(self):
        super(LinearNet, self).__init__()
        self._linear = nn.Linear(784, 10)

    def forward(self, x):
        return self._linear(x)

layer = LinearNet()

# configure model inputs
x_spec = InputSpec([None, 784], 'float32', 'x')

# convert model to inference mode
layer.eval()

save_path = 'onnx.save/linear_net'
p2o.dygraph2onnx(layer, save_path + '.onnx', input_spec=[x_spec])

# when paddlepaddle>2.0.0, you can try:
# paddle.onnx.export(layer, save_path, input_spec=[x_spec])

```

#### IPython tutorials

- [Convert to ONNX from dynamic computational graph](examples/tutorial_dygraph2onnx.ipynb)

## Documents

- [model zoo](docs/en/model_zoo.md)
- [op list](docs/en/op_list.md)
- [update notes](docs/en/change_log.md)

## License
[Apache-2.0 license](https://github.com/PaddlePaddle/paddle-onnx/blob/develop/LICENSE).