未验证 提交 a20970b1 编写于 作者: D David Lin 提交者: GitHub

添加传参

在 298 行调用 get_images_from_dir 时,文件后缀忘传了,导致infer脚本只能推理 jpg或jpeg 格式的图片
上级 92218268
......@@ -295,7 +295,7 @@ class Predictor:
def run(deploy_conf, imgs_dir, support_extensions=".jpg|.jpeg"):
# 1. scan and get all images with valid extensions in directory imgs_dir
imgs = get_images_from_dir(imgs_dir)
imgs = get_images_from_dir(imgs_dir, support_extensions)
if len(imgs) == 0:
print("No Image (with extensions : %s) found in [%s]" %
(support_extensions, imgs_dir))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册