TensorflowのPreProcessでTensorflow以外の関数を使う - Qiita

はじめに tensorflow.data.Datasetを使っていて、preprocess内でnumpyのFFTやopenCVを使ったらエラーが出て困った人向けです。いや自分自身が困ったので、そのTipです。 時間がない人向け tf.py_functionを使用すると出来る。 細かい説明 tf.data.Datasetの簡単な動き こちらはtensorflowの公式のドキュメントで書いてある通りに、datasetのloaderを作ります。pytorchでいえばdataloaderですね。 def preprocess_image(image): image = tf.image.decode_jpeg(image, channels=3) image = tf.image.resize(image, [192, 192]) return image def load_and_preproc