python - Kerras : Found 39 images belonging to 3 classes. 找到属于 3 个类的 49 个图像

Question

from keras.preprocessing.image import ImageDataGenerator

WIDTH = 299
HEIGHT = 299
BATCH_SIZE = 32

# data prep
train_datagen = ImageDataGenerator(
    preprocessing_function=preprocess_input,
    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')

validation_datagen = ImageDataGenerator(
    preprocessing_function=preprocess_input,
    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')

train_generator = train_datagen.flow_from_directory(
    TRAIN_DIR,
    target_size=(HEIGHT, WIDTH),
        batch_size=BATCH_SIZE,
        class_mode='categorical')

validation_generator = validation_datagen.flow_from_directory(
    TEST_DIR,
    target_size=(HEIGHT, WIDTH),
    batch_size=BATCH_SIZE,
    class_mode='categorical')

对于这部分代码，我的输出应该是“Found 39 images belonging to 2 classes.

找到属于 2 个类的 49 个图像。但我得到 3 个类。

请帮助我。提前致谢。还有谁能告诉我它是根据什么划分的？

完整代码请查看此链接 https://colab.research.google.com/drive/18AN2AUM5sEsTMGUzFUL0FLSULtXF4Ps0

Answer 1

Colab 在 gdrive 的工作目录中自动创建一个名为“.ipynb_checkpoints”的文件夹。确保在开始生成数据之前将其删除。