Как загрузить пользовательские данные в tfds для примера keras cyclegan?

#python-3.x #keras #deep-learning #tensorflow-datasets

Вопрос:

Согласно примеру в https://keras.io/examples/generative/cyclegan/, ранее существующий набор данных был загружен для реализации. Я пытаюсь добавить свой набор данных.

 import tensorflow_datasets as tfds
data = tfds.folder_dataset.ImageFolder('Images', shape=(256, 256, 3))
ds = data.as_dataset()
 

где «Изображения» — это корневая папка, содержащая две подпапки train и test. папка train , содержащая trainA и trainB, тест, содержащий testA и TestB.

Однако я не могу понять, как получить доступ к trainA , trainB , testA и TestB, чтобы он был принят примером keras cyclegan.