diff --git a/README.md b/README.md index 58b57f13ff2d14ca97b14e6b1ba088cde3187790..a84c59b7bf3c17d4112d02af053f86afce994a35 100644 --- a/README.md +++ b/README.md @@ -38,11 +38,11 @@ x2paddle --framework=onnx --model=onnx_model.onnx --save_dir=pd_model ### 参数选项 | 参数 | | |----------|--------------| -|--framework | 源模型类型 (tensorflow、caffe) | +|--framework | 源模型类型 (tensorflow、caffe、onnx) | |--prototxt | 当framework为caffe时,该参数指定caffe模型的proto文件路径 | |--weight | 当framework为caffe时,该参数指定caffe模型的参数文件路径 | |--save_dir | 指定转换后的模型保存目录路径 | -|--model | 当framework为tensorflow时,该参数指定tensorflow的pb模型文件路径 | +|--model | 当framework为tensorflow/pmmx时,该参数指定tensorflow的pb模型文件或onnx模型路径 | |--caffe_proto | [可选]由caffe.proto编译成caffe_pb2.py文件的存放路径,当存在自定义Layer时使用,默认为None | ## 使用转换后的模型