Учебник TF-GAN от Google

Question

Я изучаю учебник по GAN от Google . В этой записной книжке они определили input_fn , в который набор данных MNIST загружается с использованием tfds. Я сгенерировал свой собственный набор данных и сохранил его в массиве numpy (форма: 4500, 512, 512).

Я не могу понять, как работает input_fn икак я могу изменить его так, чтобы я мог вводить данные тренировок из моего gdrive, а не загружать их из наборов данных tf. Я заметил, что input_fn также используется во время обучения, когда вызывается gan_estimator.train . Кто-нибудь может объяснить, как работает эта функция?

Answer 1

Функция input_fn использует наборы данных TensorFlow в следующей строке для загрузки MNIST.

tfds.load('mnist', split=split)
               .map(_preprocess)
               .cache()
               .repeat()

Вам необходимо понять, как работают наборы данных TensorFlow, чтобы создать свой собственный, следуя требуемой структуре длябыть обработан.

Вы можете получить больше информации здесь .