TensorFlow加载本地文件夹图像形成迭代器

# flower数据集路径
PATH = '../../Data/flower'
# 训练集路径，在总路径后拼接train
train_dir = os.path.join(PATH, 'train')
# 验证集路径，在总路径后拼接val
validation_dir = os.path.join(PATH, 'val')

# 每个批次数据大小，这里将BATCH_SIZE调大为了一次读取，在训练时在分batch，或者这里小一点
# 形成一个迭代器，然后按照遍历的方式进行训练
BATCH_SIZE = 32

# 训练集，这是个迭代器
train_dataset = image_dataset_from_directory(train_dir,
shuffle=True,

TensorFlow加载本地文件夹图像形成迭代器最先出现在Python成神之路。