diff --git a/deploy/utils/get_image_list.py b/deploy/utils/get_image_list.py
index 4cef64166139ffecd834bd62e98c1b450d0478b6..0f42f36f27d8c9796fe511b05abd8f186752f5fe 100644
--- a/deploy/utils/get_image_list.py
+++ b/deploy/utils/get_image_list.py
@@ -27,9 +27,10 @@ def get_image_list(img_file):
     if os.path.isfile(img_file) and img_file.split('.')[-1] in img_end:
         imgs_lists.append(img_file)
     elif os.path.isdir(img_file):
-        for single_file in os.listdir(img_file):
-            if single_file.split('.')[-1] in img_end:
-                imgs_lists.append(os.path.join(img_file, single_file))
+        for root, dirs, files in os.walk(img_file):
+            for single_file in files:
+                if single_file.split('.')[-1] in img_end:
+                    imgs_lists.append(os.path.join(root, single_file))
     if len(imgs_lists) == 0:
         raise Exception("not found any img file in {}".format(img_file))
     imgs_lists = sorted(imgs_lists)
diff --git a/ppcls/data/utils/get_image_list.py b/ppcls/data/utils/get_image_list.py
index 6f10935ad37f180449dc93877701c42679dbd094..34d12c3c9c351870f1fc6ad1c798bb8e0d894e5f 100644
--- a/ppcls/data/utils/get_image_list.py
+++ b/ppcls/data/utils/get_image_list.py
@@ -27,9 +27,10 @@ def get_image_list(img_file):
     if os.path.isfile(img_file) and img_file.split('.')[-1] in img_end:
         imgs_lists.append(img_file)
     elif os.path.isdir(img_file):
-        for single_file in os.listdir(img_file):
-            if single_file.split('.')[-1] in img_end:
-                imgs_lists.append(os.path.join(img_file, single_file))
+        for root, dirs, files in os.walk(img_file):
+            for single_file in files:
+                if single_file.split('.')[-1] in img_end:
+                    imgs_lists.append(os.path.join(root, single_file))
     if len(imgs_lists) == 0:
         raise Exception("not found any img file in {}".format(img_file))
     imgs_lists = sorted(imgs_lists)