首页 > 解决方案 > 如何从图像路径列表创建 tf.data.Dataset

问题描述

我想创建 td.data.Dataset 实例,并且我知道输入和输出图像的路径(它是一个图像到图像的神经网络)。我知道我可以使用创建 tf.data.Datasettf.data.Dataset.list_files()但这不是我的情况,因为我需要专门使用在一个文本文件中指定的 train-val-test 拆分,并且将图像移动到特定文件夹也很乏味并且似乎是一种低效的方式来完成这项任务。

标签: pythontensorflow

解决方案


推荐阅读