Я пытаюсь использовать там код: https://www.tensorflow.org/tutorials/generative/cyclegan на моих собственных данных.
В этом руководстве объект набора данных создается с помощью этого кода:
dataset, metadata = tfds.load('cycle_gan/horse2zebra',
with_info=True, as_supervised=True)
train_horses, train_zebras = dataset['trainA'], dataset['trainB']
test_horses, test_zebras = dataset['testA'], dataset['testB']
Я не могу понять, как создать аналогичный объект из моих собственных данных, которые находятся в папке, например:
data / trainA
data / trainB
data / testA
data / testB
Есть ли метод, который выглядел бы как метод flow_from_directory чтобы загрузить мои данные в том же формате, что и tfds.load (...)
Что-то вроде
train_horses = foo("data/trainA")
спасибо за вашу помощь!