怎样一次性读取对应文件夹下所有符合要求的子文件夹和子文件夹下的文件列表呢?
很方便,使用glob就可以了
举个例子,想要读取文件夹中标号都是00**的四位数,每个文件夹有不同格式的图片,如png,jpg,我们怎么只选择读取png格式的图片呢?
import glob
dataset_dir = "your path to dataset"
# read all the png
img_list_png = glob.glob(dataset_dir+"images/00*/*.png")
# read all the jpg
img_list_jpg = glob.glob(dataset_dir+"images/00*/*.jpg")
glob.glob()可以返回符合对应模式的文件列表,用起来很方便!