From 25040fe8bc9c28e6345e43daed179c98b1fd37d8 Mon Sep 17 00:00:00 2001 From: SunAhong1993 <48579383+SunAhong1993@users.noreply.github.com> Date: Fri, 2 Aug 2019 10:31:11 +0800 Subject: [PATCH] Update README.md --- README.md | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index d6fc3d8..6899db1 100644 --- a/README.md +++ b/README.md @@ -20,16 +20,17 @@ x2paddle --framework=tensorflow --model=tf_model.pb --save_dir=pd_model ``` ### Caffe ``` -x2paddle --framework=caffe --proto=deploy.proto --weight=deploy.caffemodel --save_dir=pd_model +x2paddle --framework=caffe --prototxt=deploy.proto --weight=deploy.caffemodel --save_dir=pd_model ``` ### 参数选项 | 参数 | | |----------|--------------| |--framework | 源模型类型 (tensorflow、caffe) | -|--proto | 当framework为caffe时,该参数指定caffe模型的proto文件路径 | +|--prototxt | 当framework为caffe时,该参数指定caffe模型的proto文件路径 | |--weight | 当framework为caffe时,该参数指定caffe模型的参数文件路径 | |--save_dir | 指定转换后的模型保存目录路径 | |--model | 当framework为tensorflow时,该参数指定tensorflow的pb模型文件路径 | +|--caffe_proto | 由caffe.proto编译成caffe_pb2.py文件的存放路径,当没有安装caffe或者使用自定义Layer时使用 | ## 使用转换后的模型 转换后的模型包括`model_with_code`和`inference_model`两个目录。 -- GitLab