Augmentor を使って機械学習用の画像を水増し

これはなに?

機械学習用の画像を水増し (Image augmentation) するスクリプトです。

コード

使い方

依存モジュールをインストール。

mdbloice/Augmentor

$ pip install Augmentor

images_dir に水増し元画像たちが存在するディレクトリパスを指定。

$ vim image_augmentation_with_augmentor.py

#!/usr/bin/env python3
import Augmentor

images_dir = "./ganmodoki"
.
.
.

実行。

$ python image_augmentation_with_augmentor.py

output というディレクトリが掘られて水増し画像が保存される。

output/xxxxxxxxx.jpg
output/yyyyyyyyy.jpg
.
.
.

参考

Augmentor — Augmentor 0.2.2 documentation