TensorFlow Python - есть ли способ вставить набор данных tenorflow_datasets в ImageGenerator? - PullRequest
Новая
       19

TensorFlow Python - есть ли способ вставить набор данных tenorflow_datasets в ImageGenerator?

0 голосов
/ 26 октября 2019

Я долго пытался использовать увеличение данных в функции tenorflow.keras.preprocessing.image.ImageGenerator, но каждый пример, который я видел, передается в каталог с файлами. Моя цель - использовать tenorflow_datasets для импорта MNIST, а затем передать его в функцию дополнения данных, но я не смог выяснить, как это сделать.

Я готов использовать каталог, если он проще и еслилюбой может найти простой способ сделать это и успешно объяснить мне, как это сделать.

См. код ниже

Спасибо, Макс 

import tensorflow as tf
import tensorflow_datasets as tfds
from tensorflow.keras.preprocessing.image import ImageDataGenerator

def main():
    data, info = tfds.load("mnist", with_info=True)
    train_data, test_data = data['train'], data['test']

    image_gen = ImageDataGenerator(
        rescale=1./255,
        rotation_range=20,
        width_shift_range=0.2,
        height_shift_range=0.2,
        shear_range=0.2,
        zoom_range=0.2,
        fill_mode='nearest')

    #
    # What Do I Do Here??
    #

    # train_data_gen = image_gen.flow(data)


if __name__ == "__main__":
    main()
...