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

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