未验证 提交 b3779f3b 编写于 作者: J Jason 提交者: GitHub

Update README.md

上级 3991af9d
......@@ -33,7 +33,7 @@ python src/convert.py --pb_file tf_model.pb \
```
### 参数说明
|tf2fluid参数|说明|
|------------------|-----------------------------------------------|
|-----------|-----------------------------------------------|
|meta_file|TensorFlow模型序列化后保存的meta文件|
|ckpt_dir|TensorFlow模型保存checkpoint目录|
|pb_file|Tensorflow保存的pb格式模型|
......@@ -42,6 +42,7 @@ python src/convert.py --pb_file tf_model.pb \
|input_format|输入数据格式,当为CV模型时,可选择NHWC/NCHW/OTHER|
|output_nodes|输出tensor名,多个输出时以空格分隔|
|save_dir|转换后的模型保存路径|
目前支持tensorflow保存的checkpoint模型和将参数及模型结构序列化存储的pb模型,前者须指定meta_file和ckpt_file,后者则指定pb_file
### 转换后模型文件说明
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册