Python遍历文件夹下的所有图片

import os
import cv2

def getFileList(dir, Filelist, ext=None):
    """
    获取文件夹及其子文件夹中文件列表
    输入 dir：文件夹根目录
    输入 ext: 扩展名
    返回： 文件路径列表
    """
    newDir = dir
    if os.path.isfile(dir):
        if ext is None:
            Filelist.append(dir)
        else:
            if ext in dir[-3:]: #jpg为-3/py为-2
                Filelist.append(dir)

    elif os.path.isdir(dir):
        for s in os.listdir(dir):
            newDir = os.path.join(dir, s)
            getFileList(newDir, Filelist, ext)
    return Filelist
org_img_folder = r'./dir'

# 检索文件
imglist = getFileList(org_img_folder, [], 'jpg')
print('本次执行检索到 ' + str(len(imglist)) + ' 个jpg文件\n')

for imgpath in imglist:
    imgname = os.path.splitext(os.path.basename(imgpath))[0]
    print(imgpath)
    img = cv2.imread(imgpath, cv2.IMREAD_COLOR)
    cv2.imshow("img",img)
    cv2.waitKey(0)