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

3
X2Paddle可以将caffe、tensorflow、onnx模型转换成Paddle支持的模型。目前支持版本为caffe 1.0;tensorflow 1.x,推荐1.4.0;ONNX 1.6.0,OpSet支持 9, 10, 11版本。
4

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

9 10 11

## 多框架支持

12
|模型 | caffe | tensorflow | onnx |
13
|---|---|---|---|
14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29
|mobilenetv1 | Y | Y |  |
|mobilenetv2 | Y | Y | Y |
|resnet18 | Y | Y |  |
|resnet50 | Y | Y | Y |
|mnasnet | Y | Y |  |
|efficientnet | Y | Y | Y |
|squeezenetv1.1 | Y | Y | Y |
|shufflenet | Y | Y |  |
|mobilenet_ssd | Y | Y |  |
|mobilenet_yolov3 |  | Y |  |
|inceptionv4 |  |  |  |
|mtcnn | Y | Y |  |
|facedetection | Y |  |  |
|unet | Y | Y |  |
|ocr_attention |  |  |  |
|vgg16 |  |  |  |
30 31


32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71
## 安装

```
pip install x2paddle
```

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

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

## 使用

### Caffe

```
x2paddle --framework caffe \
         --prototxt model.proto \
	 --weight model.caffemodel \
         --save_dir paddle_model
```

### TensorFlow

```
x2paddle --framework tensorflow \
	 --model model.pb \
	 --save_dir paddle_model
```

## 转换结果说明

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

## 问题反馈

X2Paddle使用时存在问题时,欢迎您将问题或Bug报告以[Github Issues](https://github.com/PaddlePaddle/X2Paddle/issues)的形式提交给我们,我们会实时跟进。