使用 imgaug 创建和保存增强图像时遇到问题答案

【问题标题】：trouble creating and saving augmented images using imgaug使用 imgaug 创建和保存增强图像时遇到问题
【发布时间】：2020-01-10 20:33:25
【问题描述】：

使用 python v 3.7.3、pytorch v 0.4.1、imgaug 0.3.0、windows 10、Jupyter Notebook

我正在尝试遍历包含图像的多个文件夹，将每个图像增强 6 次，然后将每个增强图像的硬拷贝保存在该文件夹中。我正在使用imgaug 库来扩充图像。

我能够遍历文件夹，并使用以下代码扩充和显示文件夹内的图像：

for folder in os.listdir(path):
    for i in os.listdir(path + '\\' + folder):
        img = imageio.imread(path + '\\' + folder + '\\' + i)
        print('Original:')
        ia.imshow(img)
        img_aug = seq.augment_image(img)
        print('Augmented:')
        ia.imshow(img_aug)

但是，我希望最终将每个图像扩充 6 次，并为每个图像创建 6 个新的硬文件。我正在尝试使用this tutorial 进行这些更改。现在，我只是在尝试保存增强图像的硬拷贝。使用此代码：

for folder in os.listdir(path):
    for i in os.listdir(path + '\\' + folder):
        img = imageio.imread(path + '\\' + folder + '\\' + i)
        print('Original:')
        ia.imshow(img)
        img_aug = seq.augment_image(img)
        print('Augmented:')
        ia.imshow(img_aug)
for im, im_aug in enumerate(img_aug):
    imageio.imwrite(os.path.join(path, path + '\\' + folder + '\\' + folder + "%06d.png" % (im)), im_aug)

当我在 Jupyter 实验室打印增强图像时，它们会正常显示，但它们被保存为完全平坦的硬拷贝。它还保存了数百张这样的图片：

为什么我的图像会在 Jupyter Labs 中正确显示，但在我尝试保存硬拷贝时却以该格式保存？

【问题讨论】：

标签： python machine-learning data-augmentation

【解决方案1】：

Answered here，需要自增：

for folder in os.listdir(path):
    i = 0
    for fname in os.listdir(path + '\\' + folder):
        img = imageio.imread(path + '\\' + folder + '\\' + fname)
        print('Original:')
        ia.imshow(img)
        img_aug = seq.augment_image(img)
        print('Augmented:')
        ia.imshow(img_aug)

        imageio.imwrite(os.path.join(path, path + '\\' + folder + '\\' + folder + "%06d.png" % (i,)), img_aug)
        i += 1

【讨论】：