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

Update README.md

上级 b134d630
......@@ -81,8 +81,8 @@ model.save_inference_model("new_model_dir")
|use_cuda|转换过程中是否使用GPU,默认True|
|save_dir|转换后的模型保存路径|
目前支持tensorflow保存的checkpoint模型和将参数及模型结构序列化存储的pb模型,前者须指定meta_file和ckpt_dir,后者则指定pb_file
FAQ:输入tensor名和输出tensor名是什么?
目前支持tensorflow保存的checkpoint模型和将参数及模型结构序列化存储的pb模型,前者须指定meta_file和ckpt_dir,后者则指定pb_file
**FAQ:输入tensor名和输出tensor名是什么? **
TensorFlow模型在infer时,一般调用代码形如`sess.run([output], {input:data})`,其中output即为输出tensor,input则为输入tensor,在进行模型转换时,需提供这input和output对应的`tensor name`,如在[vgg_translate_tutorial](vgg_translate_tutorial.ipynb)中转换VGG_16模型,输入的tensor名为 "inputs", 输出的tensor名为 "vgg_16/fc8/squeezed"
### 转换后模型文件说明
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册