Keras imagedatagenerator 一括処理

Question

kerasで画像判定に関する学習を行なっています。

imagedatageneratorを使ってディレクトリ内の画像に対して一括で処理を行いたいと考えております。

現状は増やしたい画像をimageに入れ、水増し後に元のディレクトリに戻す作業をしています。

imageに画像を入れなくともtrainに入っている全ディレクトリ内の.pngに対して水増しを行いたいのですがどのように書き加えたらいいのか教えてください。

train
/dirA
/A.png
B.png
C.png
/dirB
/D.png
E.png
F.png
/dirC
/G.png
H.png
Y.png

といったディレクトリ構成になっております

import os
import glob
import numpy as np
from keras.preprocessing.image import ImageDataGenerator, load_img, img_to_array, array_to_img

def draw_images(generator, x, dir_name, index):
    save_name = 'extened' + str(index)
    g = generator.flow(x, batch_size=1, save_to_dir=output_dir, save_prefix=save_name, save_format='png')

    for i in range(10):##############拡張枚数###################
        bach = g.next()

input_dir = 'dir/a'
if __name__ == '__main__':

    ##############出力先###################
    output_dir = input_dir

    ##############入力ディレクトリ###################
    images = glob.glob("/Users/a/Desktop/image/*.png")
    
    
    ##############オーギュメンテーションの設定###################
    generator = ImageDataGenerator(
                    #zoom_range=0.8,
                    )
    
    
    # 読み込んだ画像を順に拡張
    for i in range(len(images)):
        img = load_img(images[i])
        # 画像を配列化して転置a
        x = img_to_array(img)
        x = np.expand_dims(x, axis=0)
        # 画像の拡張
        draw_images(generator, x, output_dir, i)
        
print("fin")

宜しくお願い致します。

Comments

[tiitoi]

現状のコードで変換して保存するということが実現できていそうですが、なにか問題があるのでしょうか？

[22Go]

dirAの中の.pngを見つけたら水増しをして、その後、次のdirBにアクセスして.pngを水増ししたいと思ってます。
現在は別でディレクトリを準備して水増し後に元のディレクトリに全て写している状況なんですが、もっと簡単にできたらと思いまして。。。
ちなみに現在はDesktop/train/dirA.dirB.dirCとなっており、Desktop/imageというディレクトリを作成してあります。

Answer 1

データセットのディレクトリで各クラスごとに画像がフォルダで分けられている場合は以下のようにすれば、データセットを水増ししてディレクトリに保存できます。

元の画像と水増しした画像が区別つくように、output_dir に元のデータセットと同じディレクトリ構造で保存するようにしています。

from pathlib import Path
from keras.preprocessing.image import (
    ImageDataGenerator,
    load_img,
    img_to_array,
    array_to_img,
)


# データセットのパス
dataset_dir = Path("/data/flower_photos")

# 以下のようにクラスごとにサブディレクトリがある前提
# /data/flower_photos
# |-- daisy
# |-- dandelion
# |-- roses
# |-- sunflowers
# `-- tulips

# 出力ディレクトリのパス
output_dir = Path("extend")
output_dir.mkdir(exist_ok=True)

# 1枚あたりの生成枚数 (例: 2の場合、1つの画像を2倍に水増しする)
n = 1

# 画像データ生成器を作成する。
params = {"rotation_range": 20, "width_shift_range": 0.4}  # パラメータは適当に変えてください。
datagen = ImageDataGenerator(**params)

for sub_dir in dataset_dir.iterdir():
    print(f"processing {sub_dir} ...")

    output_subdir = output_dir / sub_dir.name
    output_subdir.mkdir(exist_ok=True)

    for img_path in sub_dir.glob("*.jpg"):  # 拡張子は適当に変えてください
        # 画像を読み込む。
        img = load_img(img_path)
        x = img_to_array(img)  # PIL => numpy
        x = np.expand_dims(x, axis=0)  # (H, W, C) -> (1, H, W, C)

        gen = datagen.flow(
            x,
            batch_size=1,
            save_to_dir=output_subdir,
            save_prefix=f"extened_{img_path.stem}",
            save_format="png",
        )

        # ジェネレーターを n 回呼び出す。
        [gen.next() for i in range(n)]