模型转换工具 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的形式提交给我们，我们会实时跟进。