Skip to content

P
PaddleSeg

PaddlePaddle / PaddleSeg

通知 289
Star 8
Fork 1
P
PaddleSeg
未验证 提交 d506d5f5 编写于 作者: B Bin Long 提交者： GitHub
浏览文件
操作

Update infer.py

上级 2956a4f1
隐藏空白更改
内联 并排
Showing with 2 addition and 1 deletion
deploy/python/infer.py
浏览文件 @ d506d5f5
...@@ -31,6 +31,7 @@ gflags.DEFINE_string("conf", default="", help="Configuration File Path") ...@@ -31,6 +31,7 @@ gflags.DEFINE_string("conf", default="", help="Configuration File Path")
gflags.DEFINE_string("input_dir", default="", help="Directory of Input Images") gflags.DEFINE_string("input_dir", default="", help="Directory of Input Images")
gflags.DEFINE_boolean("use_pr", default=False, help="Use optimized model") gflags.DEFINE_boolean("use_pr", default=False, help="Use optimized model")
gflags.DEFINE_string("trt_mode", default="", 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 gflags.FLAGS = gflags.FLAGS
...@@ -315,4 +316,4 @@ if __name__ == "__main__": ...@@ -315,4 +316,4 @@ if __name__ == "__main__":
"Invalid trt_mode [%s], only support[int8, fp16, fp32]" % trt_mode) "Invalid trt_mode [%s], only support[int8, fp16, fp32]" % trt_mode)
exit(-1) exit(-1)
# run inference # run inference
run(gflags.FLAGS.conf, gflags.FLAGS.input_dir) run(gflags.FLAGS.conf, gflags.FLAGS.input_dir, gflags.FLAGS.ext)
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或