1 回答

TA貢獻(xiàn)1818條經(jīng)驗(yàn) 獲得超7個(gè)贊
您可以使用 ImageDataGenerator 的驗(yàn)證 split 關(guān)鍵字自動(dòng)拆分訓(xùn)練數(shù)據(jù)和測(cè)試數(shù)據(jù)。
train_datagen = ImageDataGenerator(rescale=1./255,
shear_range=0.2,
zoom_range=0.2,
horizontal_flip=True,
validation_split=0.2) # set validation split
train_generator = train_datagen.flow_from_directory(
train_data_dir,
target_size=(img_height, img_width),
batch_size=batch_size,
class_mode='binary',
subset='training') # set as training data
validation_generator = train_datagen.flow_from_directory(
train_data_dir, # same directory as training data
target_size=(img_height, img_width),
batch_size=batch_size,
class_mode='binary',
subset='validation') # set as validation data
model.fit_generator(
train_generator,
steps_per_epoch = train_generator.samples // batch_size,
validation_data = validation_generator,
validation_steps = validation_generator.samples // batch_size,
epochs = nb_epochs)
當(dāng)ImageDataGenerator自動(dòng)打亂您的輸入數(shù)據(jù)時(shí),您使用ImageDataGenerator的數(shù)據(jù)會(huì)被打亂和分割。
在你的情況下,你需要flow而不是flow_from_directory
添加回答
舉報(bào)