image.py 文件源码

def save_image(self, tensor, name='default.jpg'):
        image = tensor.clone().cpu()
        image = image.view(3, image_size, image_size)
        image = self.un_loader(image)
        print(image)
        try:
            im = Image.fromarray(image)
            im.save(name)
            print('image saved.')
        except Exception as e:
            print(e)
            mimage.imsave(name, image)
            print('image saved.')