Keras ImageDataGenerator столкновения
В следующем коде, когда я установил max_batches
до 2, я получаю в два раза больше общего количества оригинальных изображений, сохраненных в save_to_dir
дорожка. Чем выше я установил max_batches
за 2, больше пропадают. Например, настройка max_batches
3 дает мне примерно на 5 меньше (в зависимости от семян), чем то, что 3*n_images
равно. Я думаю, что это из-за столкновений с именами файлов? Что представляет собой автоматически сгенерированное число в конце имени файла (например, rotated_1_6896.png
), и есть ли способ настроить это, чтобы избежать этих столкновений?
datagen = ImageDataGenerator(preprocessing_function=img_preprocessor,
rotation_range=180,
width_shift_range=0.2,
height_shift_range=0.2)
# fit parameters from data
datagen.fit(X_train)
batch_size = n_images
max_batches = 3
batches_iter = 0
for X_batch, y_batch in datagen.flow(X_train, y_train,
seed=0,
batch_size=batch_size,
save_to_dir=augment_path,
save_prefix='rotated',
save_format='png',
shuffle=False):
batches_iter += 1
if batches_iter >= max_batches:
break