Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
qq_38905368
tensorflow
提交
7d8ce2ee
T
tensorflow
项目概览
qq_38905368
/
tensorflow
与 Fork 源项目一致
从无法访问的项目Fork
通知
5
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
T
tensorflow
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
7d8ce2ee
编写于
10月 20, 2016
作者:
A
A. Unique TensorFlower
提交者:
TensorFlower Gardener
10月 20, 2016
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Fix freeze_graph.py and freeze_graph_test.py to work with Saver V1 and V2.
Change: 136743848
上级
8a0e81d8
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
13 addition
and
7 deletion
+13
-7
tensorflow/python/tools/freeze_graph.py
tensorflow/python/tools/freeze_graph.py
+2
-1
tensorflow/python/tools/freeze_graph_test.py
tensorflow/python/tools/freeze_graph_test.py
+11
-6
未找到文件。
tensorflow/python/tools/freeze_graph.py
浏览文件 @
7d8ce2ee
...
...
@@ -80,7 +80,8 @@ def freeze_graph(input_graph, input_saver, input_binary, input_checkpoint,
print
(
"Input saver file '"
+
input_saver
+
"' does not exist!"
)
return
-
1
if
not
tf
.
gfile
.
Glob
(
input_checkpoint
):
# 'input_checkpoint' may be a prefix if we're using Saver V2 format
if
not
tf
.
train
.
checkpoint_exists
(
input_checkpoint
):
print
(
"Input checkpoint '"
+
input_checkpoint
+
"' doesn't exist!"
)
return
-
1
...
...
tensorflow/python/tools/freeze_graph_test.py
浏览文件 @
7d8ce2ee
...
...
@@ -27,7 +27,7 @@ from tensorflow.python.tools import freeze_graph
class
FreezeGraphTest
(
test_util
.
TensorFlowTestCase
):
def
testFreezeGraph
(
self
):
def
_testFreezeGraph
(
self
,
saver_write_version
):
checkpoint_prefix
=
os
.
path
.
join
(
self
.
get_temp_dir
(),
"saved_checkpoint"
)
checkpoint_state_name
=
"checkpoint_state"
...
...
@@ -44,9 +44,9 @@ class FreezeGraphTest(test_util.TensorFlowTestCase):
sess
.
run
(
init
)
output
=
sess
.
run
(
output_node
)
self
.
assertNear
(
2.0
,
output
,
0.00001
)
saver
=
tf
.
train
.
Saver
(
write_version
=
tf
.
train
.
SaverDef
.
V1
)
saver
.
save
(
sess
,
checkpoint_prefix
,
global_step
=
0
,
latest_filename
=
checkpoint_state_name
)
saver
=
tf
.
train
.
Saver
(
write_version
=
saver_write_version
)
checkpoint_path
=
saver
.
save
(
sess
,
checkpoint_prefix
,
global_step
=
0
,
latest_filename
=
checkpoint_state_name
)
tf
.
train
.
write_graph
(
sess
.
graph
,
self
.
get_temp_dir
(),
input_graph_name
)
# We save out the graph to disk, and then call the const conversion
...
...
@@ -54,7 +54,6 @@ class FreezeGraphTest(test_util.TensorFlowTestCase):
input_graph_path
=
os
.
path
.
join
(
self
.
get_temp_dir
(),
input_graph_name
)
input_saver_def_path
=
""
input_binary
=
False
input_checkpoint_path
=
checkpoint_prefix
+
"-0"
output_node_names
=
"output_node"
restore_op_name
=
"save/restore_all"
filename_tensor_name
=
"save/Const:0"
...
...
@@ -62,7 +61,7 @@ class FreezeGraphTest(test_util.TensorFlowTestCase):
clear_devices
=
False
freeze_graph
.
freeze_graph
(
input_graph_path
,
input_saver_def_path
,
input_binary
,
input_
checkpoint_path
,
input_binary
,
checkpoint_path
,
output_node_names
,
restore_op_name
,
filename_tensor_name
,
output_graph_path
,
clear_devices
,
""
)
...
...
@@ -84,5 +83,11 @@ class FreezeGraphTest(test_util.TensorFlowTestCase):
output
=
sess
.
run
(
output_node
)
self
.
assertNear
(
2.0
,
output
,
0.00001
)
def
testFreezeGraphV1
(
self
):
self
.
_testFreezeGraph
(
tf
.
train
.
SaverDef
.
V1
)
def
testFreezeGraphV2
(
self
):
self
.
_testFreezeGraph
(
tf
.
train
.
SaverDef
.
V2
)
if
__name__
==
"__main__"
:
tf
.
test
.
main
()
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录