Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleSeg
提交
2b47dd7d
P
PaddleSeg
项目概览
PaddlePaddle
/
PaddleSeg
通知
285
Star
8
Fork
1
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
53
列表
看板
标记
里程碑
合并请求
3
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleSeg
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
53
Issue
53
列表
看板
标记
里程碑
合并请求
3
合并请求
3
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
2b47dd7d
编写于
1月 14, 2020
作者:
B
Bin Long
提交者:
GitHub
1月 14, 2020
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Update infer.py
上级
f2a8b2b9
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
2 addition
and
2 deletion
+2
-2
deploy/python/infer.py
deploy/python/infer.py
+2
-2
未找到文件。
deploy/python/infer.py
浏览文件 @
2b47dd7d
...
...
@@ -29,7 +29,7 @@ from concurrent.futures import ThreadPoolExecutor, as_completed
gflags
.
DEFINE_string
(
"conf"
,
default
=
""
,
help
=
"Configuration File Path"
)
gflags
.
DEFINE_string
(
"input_dir"
,
default
=
""
,
help
=
"Directory of Input Images"
)
gflags
.
DEFINE_boolean
(
"use_pr"
,
default
=
Fals
e
,
help
=
"Use optimized model"
)
gflags
.
DEFINE_boolean
(
"use_pr"
,
default
=
Tru
e
,
help
=
"Use optimized model"
)
gflags
.
DEFINE_string
(
"trt_mode"
,
default
=
""
,
help
=
"Use optimized model"
)
gflags
.
DEFINE_string
(
"ext"
,
default
=
".jpeg|.jpg"
,
help
=
"Input Image File Extensions"
)
gflags
.
FLAGS
=
gflags
.
FLAGS
...
...
@@ -307,7 +307,7 @@ if __name__ == "__main__":
gflags
.
FLAGS
(
sys
.
argv
)
if
(
gflags
.
FLAGS
.
conf
==
""
or
gflags
.
FLAGS
.
input_dir
==
""
):
print
(
"Usage: python infer.py --conf=/config/path/to/your/model "
+
"--input_dir=/directory/of/your/input/images
[--use_pr=True]
"
)
"--input_dir=/directory/of/your/input/images"
)
exit
(
-
1
)
# set empty to turn off as default
trt_mode
=
gflags
.
FLAGS
.
trt_mode
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录