数据预处理-从文件夹批量读取图片

#数据预处理-从文件夹批量读取图片
文件目录

import pathlib

#获取当前路径
data_root = pathlib.Path.cwd()

#获取指定目录下的文件路径（返回是一个列表，每一个元素是一个PosixPath对象）（将三级目录下的所有文件都获取到）
all_image_paths = list(data_root.glob(’//*’))

将PosixPath对象转为字符串
all_image_paths = [str(path) for path in all_image_paths]

#获取图片类别的名称，即存放样本图片的五个文件夹的名称

label_names = sorted(item.name for item in data_root.glob(’//’) if item.is_dir())

#将类别名称转为数值型的类标

label_to_index = dict((name, index) for index, name in enumerate(label_names))

#获取所有图片的类标

all_image_labels = [label_to_index[pathlib.Path(path).parent.name]
for path in all_image_paths]