提交 a27f15eb 编写于 作者: N nbgao

fix deploy/infer.py support_extensions parameter missing problem

上级 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.
先完成此消息的编辑!
想要评论请 注册