DataLoader中加载两种数据路径的方式，以load_dataset方法为例

作者：AllinToyou | 2024-04-12 19:20:31

踩

load_dataset

一种是原图和标签分开存储，training和test里面各自有images和manual(原图和标签)：以DRIVE数据集为例：

在这里插入图片描述
另一种，如果所给的数据集的image和label是成对出现在同一个路径下，则要用一下方式：（师兄手把手教，感谢师兄！）

load_dataset类是放在初始化部分，以下面方式呈现
class Data(Dataset):
def init
self.images, self.groundtruth = load_dataset(self.root_dir, self.train)
师兄说DataLoader是核心，还有60分钟Pytorch入门手册里面的基础知识再补一补，他说他就是看了那个，不断理解数据怎么加载。
本来想用一种合适的图清晰地表示出两种数据集存储地结构，还没找到，未完待续！

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/AllinToyou/article/detail/412267