image.py 文件源码

def __init__(self):
        self.loader = transforms.Compose([
            transforms.Scale(image_size),
            transforms.ToTensor()
        ])
        self.un_loader = transforms.ToPILImage()