x2paddle.md 1.7 KB
Newer Older
1 2
# 模型转换工具 X2Paddle

3 4 5 6 7
X2Paddle可以将caffe、tensorflow、onnx模型转换成Paddle支持的模型。目前支持版本为caffe 1.0;tensorflow 1.x,推荐1.4.0;ONNX 1.6.0,OpSet支持 9, 10, 11版本。如果您使用的是PyTorch框架,请先转换为ONNX模型之后再使用X2Paddle工具转化为Paddle模型。

[X2Paddle](https://github.com/PaddlePaddle/X2Paddle)支持将Caffe/TensorFlow/ONNX模型转换为PaddlePaddle模型。
支持的模型可参考[X2Paddle模型测试库](https://github.com/PaddlePaddle/X2Paddle/blob/develop/x2paddle_model_zoo.md)

8 9


10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26
## 安装

```
pip install x2paddle
```

安装最新版本,可使用如下安装方式

```
pip install git+https://github.com/PaddlePaddle/X2Paddle.git@develop
```

## 使用

### Caffe

```
27 28 29 30
x2paddle --framework=caffe \
         --prototxt=deploy.prototxt \
         --weight=deploy.caffemodel \
         --save_dir=pd_model
31 32 33 34 35
```

### TensorFlow

```
36 37 38 39 40 41 42 43 44 45 46
x2paddle --framework=tensorflow \
         --model=tf_model.pb \
         --save_dir=pd_model
```

### ONNX

```
x2paddle --framework=onnx \
         --model=onnx_model.onnx \
         --save_dir=pd_model
47 48 49 50 51 52 53 54 55 56
```

## 转换结果说明

在指定的`save_dir`下生成两个目录  
1. inference_model : 模型结构和参数均序列化保存的模型格式
2. model_with_code : 保存了模型参数文件和模型的python代码

## 问题反馈

57
受限于不同框架的差异,部分模型可能会存在目前无法转换的情况,如TensorFlow中包含控制流的模型,NLP模型等。对于CV常见的模型,如若您发现无法转换或转换失败,存在较大差异等问题,欢迎您将问题或Bug报告以[Github Issues](https://github.com/PaddlePaddle/X2Paddle/issues)的形式提交给我们,我们会实时跟进。