在使用flow_from_directory
时,我们可以将文件夹视为一个文件夹,并实际复制它们,通过在Keras的ImageDataGenerator中设置参数follow_links=True
来实现。
flow_from_directory
是Keras中用于从磁盘文件夹中读取图像数据的方法,通常用于在训练模型时生成批次的图像数据。默认情况下,它会遍历文件夹中的所有子文件夹,并将每个子文件夹视为一个类别。
然而,当我们希望将文件夹本身视为一个单独的类别时,并在生成批次时保留文件夹结构,可以设置follow_links=True
来实现此目的。
具体步骤如下:
from tensorflow.keras.preprocessing.image import ImageDataGenerator
follow_links=True
:datagen = ImageDataGenerator(follow_links=True)
flow_from_directory
加载数据:train_generator = datagen.flow_from_directory('path/to/directory', target_size=(224, 224), batch_size=32, class_mode='categorical')
其中,'path/to/directory'
是包含文件夹的路径,target_size
是图像的大小,batch_size
是每个批次的图像数量,class_mode
指定类别的类型。
这样,flow_from_directory
将会将文件夹视为一个单独的类别,并在生成批次时保留文件夹结构。
推荐腾讯云相关产品:腾讯云对象存储(COS)
注意:以上是根据题目要求给出的答案,没有提及亚马逊AWS、Azure、阿里云、华为云、天翼云、GoDaddy、Namecheap、Google等流行的云计算品牌商。
领取专属 10元无门槛券
手把手带您无忧上云