сохранить только изображения в наборе данных tensorflow? - PullRequest
1 голос
/ 20 июня 2020

Я пытаюсь создать AutoEncoder, используя наборы данных тензорного потока в качестве входных и выходных данных. Я загрузил набор данных tf_flowers, используя tensorflow dataset.

splits = ["train[:80%]", "train[80%:90%]", "train[90%:100%]"]`
(ds_train, ds_validation, ds_test), info = tfds.load(name="tf_flowers", with_info=True, split=splits, as_supervised=False)

. Данные загрузились идеально, как и ожидалось. Теперь я хочу изменить свои наборы данных так, чтобы у меня были только изображения, т. Е. Мои файлы и мои ярлыки должны быть моими изображениями. Как я могу изменить свой код для этого?

...