Python中图像增强的图像增强管道出错

时间:2018-06-16 10:54:48

标签: python machine-learning augmentor

据我所知,路径是正确的,我也遵循Augmentor文档。

代码:

import Augmentor
import os
import warnings
warnings.filterwarnings('ignore')
import keras
import glob

for img in glob.glob("C:\\Users\\Diganta\\Desktop\\Courses and Projects\\Projects\\Bennet\\irregular*.jpg"):
    p = Augmentor.Pipeline(img)
    p.rotate(probability=0.7, max_left_rotation=10, max_right_rotation=10)
    p.zoom(probability=0.5, min_factor=1.1, max_factor=1.5)
    p.sample(100)

这确实已经运行,但是根据Augmentor文档中指定的目录,没有在目录中创建包含增强图像的输出文件夹

1 个答案:

答案 0 :(得分:0)

我不是Augmentor专家,但通过查看source code,看起来Pipeline想要一个源目录作为参数,它会自动找到那里的所有图片。

尝试直接传递目录,没有globs和循环:

p = Augmentor.Pipeline("C:\\Users\\Diganta\\Desktop\\Courses and Projects\\Projects\\Bennet")
p.rotate(probability=0.7, max_left_rotation=10, max_right_rotation=10)
p.zoom(probability=0.5, min_factor=1.1, max_factor=1.5)
p.sample(100)

在我看来,你想要在所有图像上运行管道,而不是子集。如果是这种情况,请将p.sample(100)替换为:

p.sample(0)

或:

p.process()