Hub with Keras(Keras編)

◎環境：Python3.7.2/TensorFlow 1.13

概要

TensorFlow/Kerasのチュートリアル Hub with Kerasをやってみて、調べた内容(Keras関連)をまとめておく。

ImageDataGenerator

keras.preprocessing.image.ImageDataGeneratorクラスは、画像データのバッチの作成を行う。

Keras Documentation - ImageDataGeneratorクラス

コンストラクタ

image_generator = tf.keras.preprocessing.image.ImageDataGenerator(rescale=1/255)
# rescale : 画素値に対し、指定した値を積算する(ここでは画素値×1/255)

flow_from_directoryメソッド

image_data = image_generator.flow_from_directory(str(data_root), target_size=IMAGE_SIZE)
# flow_from_directory() : ディレクトリのパスを受け取り、拡張/正規化したデータのバッチを生成
#  - directory : ディレクトリのパス(クラスごとに画像ファイルの入ったサブディレクトリを含む)
#  - target_size : 全画像はこの値にリサイズされる(整数のタプル(height,width))