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

0 ответов

Другие вопросы по тегам