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

Update README.md

上级 c09dee40
...@@ -11,10 +11,20 @@ ...@@ -11,10 +11,20 @@
## Introduce ## Introduce
tensorflow2fluid支持将训练好的TensorFlow模型转至PaddlePaddle模型,转换后的文件list如下 tensorflow2fluid支持将训练好的TensorFlow模型转至PaddlePaddle模型,转换后的保存目录中,文件list如下表所示
> save_dir |文件|作用|
> | |------------------|-----------------------------------------------|
> |----my_model.py 基于PaddlePaddle实现的模型网络结构python代码 |my_model.py|基于PaddlePaddle实现的模型网络结构python代码|
> |----ref_name.txt my_model.py中各输出与原TensorFlow模型中的输出对应关系 |ref_name.txt|my_model.py中各tensor与原TensorFlow模型中的tensor对应关系|
> |----const_*/params_* 转换后的模型参数文件 |const_*/params_*|转换后的模型参数文件|
## Usage
使用tensorflow2fluid转换模型时,所需的信息如下
|参数|说明|
|------------------|-----------------------------------------------|
|input_nodes|输入tensor名,多个输入时以空格分隔|
|input_shape|输入tensor的shape(batch维度以None表示),shape之间以空格分隔,shape内各维度以逗号分隔,须与input_nodes对应|
|output_shape|输出tensor名,多个输出时以空格分隔|
|save_dir|转换后的模型保存路径|
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册