From b3779f3bacfc551890e52f1659a6fa5d2a351921 Mon Sep 17 00:00:00 2001 From: Jason Date: Wed, 13 Mar 2019 19:56:24 +0800 Subject: [PATCH] Update README.md --- tensorflow2fluid/README.md | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tensorflow2fluid/README.md b/tensorflow2fluid/README.md index 9f15609..52d34b6 100644 --- a/tensorflow2fluid/README.md +++ b/tensorflow2fluid/README.md @@ -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 ### 转换后模型文件说明 -- GitLab