合并 x2paddle 和 opt 的一键脚本¶
背景:如果想用 Paddle Lite 运行第三方来源(tensorflow、caffe、onnx)模型,一般需要经过两次转化。即使用 x2paddle 工具将第三方模型转化为 PaddlePaddle 格式,再使用 opt 将 PaddlePaddle 模型转化为 Padde Lite 可支持格式。 为了简化这一过程,我们提供一键脚本,将 x2paddle 转化和 opt 转化合并:
一键转化脚本:auto_transform.sh
环境要求:使用auto_transform.sh
脚本转化第三方模型时,需要先安装 x2paddle 环境,请参考 x2paddle 环境安装方法 安装 x2paddle 和 x2paddle 依赖项( tensorflow、caffe 等)。
使用方法:
(1)打印帮助帮助信息:sh ./auto_transform.sh
,Ubuntu下请执行 bash ./auto_transform.sh
(2)转化模型方法
USAGE:
auto_transform.sh combines the function of x2paddle and opt, it can
tranform model from tensorflow/caffe/onnx form into paddle-lite naive-buffer form.
----------------------------------------
example:
sh ./auto_transform.sh --framework=tensorflow --model=tf_model.pb --optimize_out=opt_model_result
----------------------------------------
Arguments about x2paddle:
--framework=(tensorflow|caffe|onnx);
--model='model file for tensorflow or onnx';
--prototxt='proto file for caffe' --weight='weight file for caffe'
For TensorFlow:
--framework=tensorflow --model=tf_model.pb
For Caffe:
--framework=caffe --prototxt=deploy.prototxt --weight=deploy.caffemodel
For ONNX
--framework=onnx --model=onnx_model.onnx
Arguments about opt:
--valid_targets=(arm|opencl|x86|x86_opencl|npu); valid targets on Paddle-Lite.
--fluid_save_dir='path to outputed model after x2paddle'
--optimize_out='path to outputed Paddle-Lite model'
----------------------------------------