前提:github下载源码编译相关工具
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模型。
pip install git+https://github.com/PaddlePaddle/X2Paddle.git@develop
--framework源模型类型 (tensorflow、caffe、onnx)
--prototxt当framework为caffe时,该参数指定caffe模型的proto文件路径
--weight当framework为caffe时,该参数指定caffe模型的参数文件路径
--save_dir指定转换后的模型保存目录路径
--model当framework为tensorflow/onnx时,该参数指定tensorflow的pb模型文件或onnx模型路径
--input_shape_dict[可选] For ONNX, 定义ONNX模型输入大小