Воспроизводимые поезда / тестовые расщепления с помощью ImageDataGenerator - PullRequest
0 голосов
/ 20 февраля 2019

Я хочу загрузить изображения с помощью ImageDataGenerator и разделить их на поезд и набор тестов с помощью параметра validation_split.Однако из документации Keras я не смог понять, как выполняется разделение (рандомизировано ли оно? Всегда ли оно выбирает одни и те же изображения?) Или как его можно засечь, чтобы воспроизвести разделение.

Этомой код до сих пор:

# Create a keras data generator
from tensorflow.keras.preprocessing.image import ImageDataGenerator

datagen = ImageDataGenerator(validation_split=0.1, rescale=1./255)
train_generator = datagen.flow_from_directory(directory='poc_train_data/data', target_size=input_size, subset='training', class_mode='binary')
val_generator = datagen.flow_from_directory(directory='poc_train_data/data', target_size=input_size, subset='validation', class_mode='binary')

class_indices = train_generator.class_indices
index_to_class = {v:k for k,v in class_indices.items()}

Как мне изменить это так, чтобы он всегда давал одно и то же случайное разбиение?

...