diff --git a/tensorflow2fluid/README.md b/tensorflow2fluid/README.md index 72d40f231f77b9f3ac006925cca933625ea1f274..42b06c5d5d54042ca0e5e85178bc8980843af2a4 100644 --- a/tensorflow2fluid/README.md +++ b/tensorflow2fluid/README.md @@ -11,10 +11,20 @@ ## Introduce -tensorflow2fluid支持将训练好的TensorFlow模型转至PaddlePaddle模型,转换后的文件list如下 +tensorflow2fluid支持将训练好的TensorFlow模型转至PaddlePaddle模型,转换后的保存目录中,文件list如下表所示 -> save_dir -> | -> |----my_model.py 基于PaddlePaddle实现的模型网络结构python代码 -> |----ref_name.txt my_model.py中各输出与原TensorFlow模型中的输出对应关系 -> |----const_*/params_* 转换后的模型参数文件 +|文件|作用| +|------------------|-----------------------------------------------| +|my_model.py|基于PaddlePaddle实现的模型网络结构python代码| +|ref_name.txt|my_model.py中各tensor与原TensorFlow模型中的tensor对应关系| +|const_*/params_*|转换后的模型参数文件| + +## Usage +使用tensorflow2fluid转换模型时,所需的信息如下 + +|参数|说明| +|------------------|-----------------------------------------------| +|input_nodes|输入tensor名,多个输入时以空格分隔| +|input_shape|输入tensor的shape(batch维度以None表示),shape之间以空格分隔,shape内各维度以逗号分隔,须与input_nodes对应| +|output_shape|输出tensor名,多个输出时以空格分隔| +|save_dir|转换后的模型保存路径|