模型转换工具 X2Paddle

X2Paddle可以将caffe、tensorflow、onnx模型转换成Paddle支持的模型。

X2Paddle支持将Caffe/TensorFlow模型转换为PaddlePaddle模型。目前X2Paddle支持的模型参考x2paddle_model_zoo

多框架支持

模型caffetensorflowonnx
mobilenetv1YY
mobilenetv2YYY
resnet18YY
resnet50YYY
mnasnetYY
efficientnetYYY
squeezenetv1.1YYY
shufflenetYY
mobilenet_ssdYY
mobilenet_yolov3Y
inceptionv4
mtcnnYY
facedetectionY
unetYY
ocr_attention
vgg16

安装

  1. pip install x2paddle

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

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

使用

Caffe

  1. x2paddle --framework caffe \
  2. --prototxt model.proto \
  3. --weight model.caffemodel \
  4. --save_dir paddle_model

TensorFlow

  1. x2paddle --framework tensorflow \
  2. --model model.pb \
  3. --save_dir paddle_model

转换结果说明

在指定的save_dir下生成两个目录

  1. inference_model : 模型结构和参数均序列化保存的模型格式
  2. model_with_code : 保存了模型参数文件和模型的python代码

问题反馈

X2Paddle使用时存在问题时,欢迎您将问题或Bug报告以Github Issues的形式提交给我们,我们会实时跟进。