python读取指定文件夹下的图片（glob获取）

定义traverse_images函数，仅需要改变下根路径即可

glob是python中用来查找符合特定规则的文件路径名的函数

import os 
from glob import glob

def traverse_images (folder_path):
    image_formats = ['*.jpg', '*.jpeg', '*.png', '*.gif'] #可根据需要添加其他图片格式
    images = []
    for image_format in image_formats:
        images.extend(glob(os.path.join(folder_path, image_format)))
    return(images)

#----使用----
# 替换为您要遍历的文件夹路径
folder_path = '/kaggle/input/ffdi-phase1-sample/ffdi_phase1_sample/valset/'
images = traverse_images(folder_path)
print(images[:10])

参考：

1. python之glob模块以及根据路径获取文件名_python glob文件名-CSDN博客【推荐】
2. 循序遍历获取指定文件夹下的图片路径(python实现) - CSDN文库