模型转换工具 X2Paddle¶
X2Paddle可以将caffe、tensorflow、onnx模型转换成Paddle支持的模型。
X2Paddle 支持将Caffe/TensorFlow模型转换为PaddlePaddle模型。目前X2Paddle支持的模型参考 x2paddle_model_zoo 。
多框架支持¶
模型 |
caffe |
tensorflow |
onnx |
---|---|---|---|
mobilenetv1 |
Y |
Y |
F |
mobilenetv2 |
Y |
Y |
Y |
resnet18 |
Y |
Y |
F |
resnet50 |
Y |
Y |
Y |
mnasnet |
Y |
Y |
F |
efficientnet |
Y |
Y |
Y |
squeezenetv1.1 |
Y |
Y |
Y |
shufflenet |
Y |
Y |
F |
mobilenet_ssd |
Y |
Y |
F |
mobilenet_yolov3 |
F |
Y |
F |
inceptionv4 |
F |
F |
F |
mtcnn |
Y |
Y |
F |
facedetection |
Y |
F |
F |
unet |
Y |
Y |
F |
ocr_attention |
F |
F |
F |
vgg16 |
F |
F |
F |
安装¶
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 下生成两个目录
inference_model : 模型结构和参数均序列化保存的模型格式
model_with_code : 保存了模型参数文件和模型的python代码
问题反馈
X2Paddle使用时存在问题时,欢迎您将问题或Bug报告以 Github Issues 的形式提交给我们,我们会实时跟进。