diff --git a/README.md b/README.md index d6fc3d871729b139faaf254b38bba97e83b867a1..6899db1e87ed558ea5c5cdd4bc07986d0f5c21ba 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`两个目录。