Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
曾经的那一瞬间
Models
提交
b729e4ec
M
Models
项目概览
曾经的那一瞬间
/
Models
12 个月 前同步成功
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
Models
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
b729e4ec
编写于
5月 13, 2022
作者:
A
A. Unique TensorFlower
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Internal change
PiperOrigin-RevId: 448644197
上级
bce895d9
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
8 addition
and
3 deletion
+8
-3
official/vision/beta/projects/yolo/serving/export_module_factory.py
...ision/beta/projects/yolo/serving/export_module_factory.py
+1
-0
official/vision/serving/export_saved_model_lib.py
official/vision/serving/export_saved_model_lib.py
+7
-3
未找到文件。
official/vision/beta/projects/yolo/serving/export_module_factory.py
浏览文件 @
b729e4ec
...
...
@@ -61,6 +61,7 @@ class ExportModule(export_base.ExportModule):
preprocessor
=
preprocessor
,
inference_step
=
inference_step
,
postprocessor
=
postprocessor
)
self
.
eval_postprocessor
=
eval_postprocessor
self
.
input_signature
=
input_signature
@
tf
.
function
...
...
official/vision/serving/export_saved_model_lib.py
浏览文件 @
b729e4ec
...
...
@@ -15,7 +15,7 @@
r
"""Vision models export utility function for serving/inference."""
import
os
from
typing
import
Optional
,
List
from
typing
import
Optional
,
List
,
Union
,
Text
,
Dict
from
absl
import
logging
import
tensorflow
as
tf
...
...
@@ -44,7 +44,8 @@ def export_inference_graph(
save_options
:
Optional
[
tf
.
saved_model
.
SaveOptions
]
=
None
,
log_model_flops_and_params
:
bool
=
False
,
checkpoint
:
Optional
[
tf
.
train
.
Checkpoint
]
=
None
,
input_name
:
Optional
[
str
]
=
None
):
input_name
:
Optional
[
str
]
=
None
,
function_keys
:
Optional
[
Union
[
List
[
Text
],
Dict
[
Text
,
Text
]]]
=
None
,):
"""Exports inference graph for the model specified in the exp config.
Saved model is stored at export_dir/saved_model, checkpoint is saved
...
...
@@ -72,6 +73,9 @@ def export_inference_graph(
will use it to read the weights.
input_name: The input tensor name, default at `None` which produces input
tensor name `inputs`.
function_keys: a list of string keys to retrieve pre-defined serving
signatures. The signaute keys will be set with defaults. If a dictionary
is provided, the values will be used as signature keys.
"""
if
export_checkpoint_subdir
:
...
...
@@ -130,7 +134,7 @@ def export_inference_graph(
export_base
.
export
(
export_module
,
function_keys
=
[
input_type
],
function_keys
=
function_keys
if
function_keys
else
[
input_type
],
export_savedmodel_dir
=
output_saved_model_directory
,
checkpoint
=
checkpoint
,
checkpoint_path
=
checkpoint_path
,
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录