Я использую ImageDataGenerator для создания набора проверки (для классификации изображений с использованием TF (Keras) из помеченного каталога изображений. Каталоги 0,1,2,3,4 соответствуют классу каждого изображения, они содержат 488 , 185, 130, 131, 91 изображений соответственно.
train_image_generator = ImageDataGenerator(rescale=1./255) # Generator for our training data
# validation_image_generator = ImageDataGenerator(rescale=1./255) # Generator for our validation data
train_data_gen = train_image_generator.flow_from_directory(batch_size=batch_size,
directory=train_dir,
shuffle=True,
target_size=(IMG_HEIGHT, IMG_WIDTH),
class_mode='categorical')
возвращает
Found 0 images belonging to 5 classes.
и код ниже
validation_data_generator = train_image_generator.flow_from_directory(
train_dir, # same directory as training data
target_size=(IMG_HEIGHT, IMG_WIDTH),
batch_size=batch_size,
class_mode='categorical',
subset='validation') # set as validation data
выводит:
Found 0 images belonging to 5 classes.
Что не так, пожалуйста? Я полагаю, что в проверочном наборе есть как минимум несколько изображений для последнего класса!
Любая помощь будет принята с благодарностью.
CS