模型转换工具 X2Paddle
X2Paddle可以将caffe、tensorflow、onnx模型转换成Paddle支持的模型。
X2Paddle支持将Caffe/TensorFlow模型转换为PaddlePaddle模型。目前X2Paddle支持的模型参考x2paddle_model_zoo。
多框架支持
模型 | caffe | tensorflow | onnx |
---|---|---|---|
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 |
安装
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的形式提交给我们,我们会实时跟进。