模型转换工具 X2Paddle
X2Paddle可以将caffe、tensorflow、onnx模型转换成Paddle支持的模型。目前支持版本为caffe 1.0;tensorflow 1.x,推荐1.4.0;ONNX 1.6.0,OpSet支持 9, 10, 11版本。如果您使用的是PyTorch框架,请先转换为ONNX模型之后再使用X2Paddle工具转化为Paddle模型。
X2Paddle支持将Caffe/TensorFlow/ONNX模型转换为PaddlePaddle模型。 支持的模型可参考X2Paddle模型测试库。
安装
pip install x2paddle
安装最新版本,可使用如下安装方式
pip install git+https://github.com/PaddlePaddle/X2Paddle.git@develop
使用
Caffe
x2paddle --framework=caffe \
--prototxt=deploy.prototxt \
--weight=deploy.caffemodel \
--save_dir=pd_model
TensorFlow
x2paddle --framework=tensorflow \
--model=tf_model.pb \
--save_dir=pd_model
ONNX
x2paddle --framework=onnx \
--model=onnx_model.onnx \
--save_dir=pd_model
转换结果说明
在指定的save_dir
下生成两个目录
- inference_model : 模型结构和参数均序列化保存的模型格式
- model_with_code : 保存了模型参数文件和模型的python代码
问题反馈
受限于不同框架的差异,部分模型可能会存在目前无法转换的情况,如TensorFlow中包含控制流的模型,NLP模型等。对于CV常见的模型,如若您发现无法转换或转换失败,存在较大差异等问题,欢迎您将问题或Bug报告以Github Issues的形式提交给我们,我们会实时跟进。