Python 获取文件夹下的所有图片

• 判断文件是否是图片:

def is_image_file(filename):
    return any(filename.endswith(extension) for extension in ['.png', '.jpg', '.jpeg', '.PNG', '.JPG', '.JPEG'])

• 从文件夹中获取图片并保存图片路径:

image_filenames = [join(dataset_dir, x) for x in listdir(dataset_dir) if is_image_file(x)]

eg:

dataset_dir='data/VOC2012/train1'
image_filenames = [join(dataset_dir, x) for x in listdir(dataset_dir) if is_image_file(x)]
for image_filename in image_filenames:
    print(image_filename)

data/VOC2012/train1/VOC2012_00009.jpg
data/VOC2012/train1/VOC2012_00002.jpg
data/VOC2012/train1/VOC2012_00007.jpg
data/VOC2012/train1/VOC2012_00006.jpg
data/VOC2012/train1/VOC2012_00008.jpg
data/VOC2012/train1/VOC2012_00005.jpg
data/VOC2012/train1/VOC2012_00004.jpg
data/VOC2012/train1/VOC2012_00001.jpg
data/VOC2012/train1/VOC2012_00003.jpg

更新:

import os
trainpath='/home/*/*/SR/raisr/test'
def get_img_file(file_name):
    imagelist = []
    for parent, dirnames, filenames in os.walk(file_name):
        for filename in filenames:
            if filename.lower().endswith(('.bmp', '.dib', '.png', '.jpg', '.jpeg', '.pbm', '.pgm', '.ppm', '.tif', '.tiff')):
                imagelist.append(os.path.join(parent, filename))
        return imagelist
   

上述函数get_img_file 返回一个包含文件路径的list列表