From a27f15ebc32b6f8b6104fc43a7b129f7dc8d509f Mon Sep 17 00:00:00 2001
From: nbgao <nbgao@126.com>
Date: Tue, 1 Sep 2020 11:13:31 +0800
Subject: [PATCH] fix deploy/infer.py support_extensions parameter missing
 problem

---
 deploy/python/infer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/deploy/python/infer.py b/deploy/python/infer.py
index 01f044e0..e1d885d2 100644
--- 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))
-- 
GitLab