目录
遍历目录下的所有目录:
import os
import shutil
dir_a=r'E:\xxx'
dir_names = ['%s' % (i[0]) for i in os.walk(dir_a) if i[0].endswith('real_images')]
for dir_name in dir_names:
print(dir_name)
筛选目录下的所有子目录下的图片:
dir_a=r'imgs'
img_files = ['%s/%s' % (i[0].replace("\\","/"), j) for i in os.walk(dir_a) for j in i[-1] if j.endswith(('jpg', 'png', 'jpeg'))]