python获取指定目录下面的图片

问题

python获取指定目录下面的图片还是比较简单的，但是代码总不能老是记录在脑子里面，于是就有了这篇“烂笔头”...

代码：

import os
def load_imgs(folder):
    '''
    加载文件夹下的图片
    '''
    imgs = []
    for img_path in os.listdir(folder):
        ext = os.path.splitext(img_path)
        if len(ext) > 1 and is_img(ext[1]):
            imgs.append(img_path)

    return imgs

def is_img(ext):
    '''
    判断文件后缀是否是图片
    '''
    ext = ext.lower()
    if ext == '.jpg':
        return True
    elif ext == '.png':
        return True
    elif ext == '.jpeg':
        return True
    elif ext == '.bmp':
        return True
    else:
        return False

总结

上述两个方法一个是扫码目录下面的文件，一个是根据文件后缀判断是否是图片