python - 文件夹上的 ImageDataGenerator

Question

我的keras文档中的代码有问题。在下面的代码中，它只加载了 1 张图像。如果我的文件夹中有 10 个图像，我将如何使用此代码，因为 load_img 函数仅加载 1 个图像，而我想加载包含 10 个图像的文件夹并对所有图像应用数据生成，有帮助吗？

from keras.preprocessing.image import ImageDataGenerator, 
array_to_img, img_to_array, load_img

datagen = ImageDataGenerator(
rotation_range=40,
width_shift_range=0.2,
height_shift_range=0.2,
shear_range=0.2,
zoom_range=0.2,
horizontal_flip=True,
fill_mode='nearest')

img = load_img('data/train/cats/cat.0.jpg') # this is a PIL image
x = img_to_array(img)
x = x.reshape((1,) + x.shape)

i = 0
for batch in datagen.flow(x, batch_size=1,
save_to_dir='preview', save_prefix='cat', save_format='jpeg'):
i += 1
if i > 20:
break

Answer 1

您可以使用 ImageDataGenerator 的 flow_from_directory 函数，keras 会自动为您完成该操作

datagen=ImageDataGenerator()
generator=datagen.flow_from_directory(directory)

编辑:
您应该具有以下文件夹结构

directory/
    class_label_1/
       image1
       image2
       .
       .
    class_label_2/
       .
       .

您的父目录应该为每个类都有一个文件夹，并且该类文件夹应该包含属于该类的图像。文件夹名称被选为标签