朴实无华的Dataset
朴实无华的Dataset自定义Dataset,该Dataset用于读取训练图像和标签图像,并做简单的尺寸调整和归一化处理,不包含其他的图像预处理操作。具体的实例化和使用方法,可以参考main.py的使用。SimpleDataset.pyimport torchimport torch.nn as nnimport mathimport torch.nn.functional as Ffrom PIL import Imagefrom torch.autograd import Variab










