From a20970b15ac6b6cd69fc460164f6521acf90c36f Mon Sep 17 00:00:00 2001
From: David Lin <davidlindev@gmail.com>
Date: Sun, 30 Aug 2020 15:03:49 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E4=BC=A0=E5=8F=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

在 298 行调用 get_images_from_dir 时，文件后缀忘传了，导致infer脚本只能推理 jpg或jpeg 格式的图片
---
 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