batch_size=batch_size, shuffle=True, num_workers=2, pin_memory=True)参考
一种是原图和标签分开存储,training和test里面各自有images和manual(原图和标签):以DRIVE数据集为例: 另一种,如果所给的数据集的image和label是成对出现在同一个路径下,则要用一下方式:(师兄手把手教,感谢师兄!) load_dataset类是放在初始化部分,以下面方式呈现 class Data(Dataset): def init self.images, se...
batch_size=batch_size, shuffle=True, num_workers=2, pin_memory=True)参考