не предсказывает правильное количество классов - PullRequest
Новая
       13

не предсказывает правильное количество классов

0 голосов
/ 04 октября 2019 
train_data_dir = mypath

Найдено 20116 изображений, принадлежащих 1 классу.

У меня есть 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

Спасибо

...