0

我正在尝试使用 Keras 的 DataImageGenerator 来增加图像以创建旋转副本。但是，在执行此操作时，一些旋转图像包含伪影，其中出现两个重复的“重影”图像，以及正确的旋转图像，并且图像的整体不透明度很低。见截图。请注意，并非所有生成的图像都会发生这种情况，仅适用于具有相同 X 且图像名称为“aug_X_ABCD.png”的一些图像。

具有适当增强的图像示例，以及具有“幽灵副本”的图像

我了解当 fill_mode 未设置为“constant”时可能会出现伪影，但在我的代码中，我已声明 fill_mode='constant' 和 cval='255'，因此应设置图像中的任何“新”区域变白。为什么会这样？

我的代码如下。谢谢！

datagen = ImageDataGenerator(fill_mode='constant', cval=255, rotation_range=90,horizontal_flip=True)

image_directory = '/content/input/'
SIZE = 128
dataset = []

my_images = os.listdir(image_directory)
for i, image_name in enumerate(my_images):
    if (image_name.split('.')[1] == 'jpg'):
        image = io.imread(image_directory + image_name)
        image = Image.fromarray(image, 'RGB')
        image = image.resize((SIZE,SIZE))
        dataset.append(np.array(image))

x = np.array(dataset)

for batch in datagen.flow(x, batch_size=16,  
                          save_to_dir='/aug', 
                          save_prefix='aug', 
                          save_format='png'):
    i += 1
    if i > 20:
4

0 回答 0