Merge pull request #363 from davidlinhl/release/v0.5.0

113b5f90 · wuzewu · GitHub · 92218268 · a20970b1 · 113b5f90
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

deploy/python/infer.py deploy/python/infer.py +1 -1

未找到文件。
--- a/deploy/python/infer.py
+++ b/deploy/python/infer.py
@@ -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))