モデルに対する拡張性のある入力の実装

[uchihashikenshi]

目的

オフラインの初期学習フェイズで画像から状態を推定するニューラルネットを作成する際、データの入出力を行う部分を実装する。

提案内容

• Iterator として実装する
• 画像を入力とするモデルとして、最低限の前処理を備えた keras.preprocessing.image.ImageDataGenerator を利用する
• __call__ で Iterator を呼ぶことでバッチに対して前処理を適用する
• endswith で複数モダリティのデータを結合できる仕様

関連