首页 > 解决方案 > 如何使用 ImageDataGenerator flow_from_directory 以原始名称作为前缀保存增强图像

问题描述

我以下列方式使用 ImageDataGenerator:

# create the imageDataGenerator
train_datagen = ImageDataGenerator(samplewise_center=gen_keras_mean_norm_sample_wise)

# flow from directory
train_generator = train_datagen.flow_from_directory(
                directory=train_dir,
                color_mode="grayscale",
                target_size=image_size,
                batch_size=batch,
                class_mode='categorical',
                interpolation='area',
                shuffle=True,
                seed=seed,
                save_to_dir=aug_save_dir_train)


# train
history = model.fit(train_generator,
                    epochs=epochs,
                    steps_per_epoch=steps_train,
                    callbacks=callbacks,
                    validation_data=validation_generator,
                    validation_steps=steps_val,
                    class_weight=None,
                    verbose=1,
                    max_queue_size=30,
                    initial_epoch=initial_epoch,
                    shuffle=True)

但是它用一个全新的名称保存了增强图像,这使得很难对原始图像和增强图像之间的差异进行完整性检查。

如何使用原始名称的前缀保存增强图像?

标签: kerasdata-generation

解决方案


推荐阅读