Google ColabのTPU環境でImageDataGeneratorライクなデータ拡張を実現する - Qiita

はじめに Keras使用者はImageDataGenaratorを使ってデータ拡張を行うことが一般的なはずだが、TPU環境ではImageDataGeneratorが直接使えない。 そこで、tf.data.Datasetにmapとして適用する関数を各種用意した。ImageDataGenaratorのほぼすべての機能に対応しているが、data_formatは'channels_last'を前提としている。 環境は、TensorFlow(2.3.0)/tf.keras(2.4.0)。 #基本機能のみ とりあえずflipとshiftだけあれば良い場合に使う。 パラメータはImageDataGeneratorと同じなので説明は省略。 ただし、fill_modeは'constant'と'reflect'しか使えない。 import tensorflow as tf @tf.function def s