不能预测正确的班级数量

Question

train_data_dir = mypath

找到了属于1类的20116张图像。

我有2个班级，但只找到一个。 我该怎么做才能解决这个问题？ 这是我的代码：

train_datagenerator = ImageDataGenerator(rescale=1./255,
    shear_range=0.2,
    zoom_range=0.2,
    horizontal_flip=True,
    validation_split=0.2) # set validation split 20% versus 80% pour training


train_generator = train_datagenerator.flow_from_directory(
    train_data_dir,
    target_size=(image_size, image_size), 
    batch_size=BATCH_SIZE_TRAINING,
    class_mode='categorical', shuffle=False, subset='training') # set as training data

validation_generator = train_datagenerator.flow_from_directory(
    train_data_dir, # same directory as training data 
    target_size=(image_size, image_size), 
    batch_size=BATCH_SIZE_TRAINING,
    class_mode='categorical', shuffle=False, subset='validation') # set as validation data

谢谢