仅作为记录,大佬请跳过。
import torchvision.transforms as transformsunloader = transforms.ToPILImage()
def tensor_to_PIL(tensor):image = tensor.cpu().clone()image = image.squeeze(0)image = unloader(image)return imageimg_t_i=tensor_to_PIL(img_t)
![在这里插入图片描述](https://img.php1.cn/3cd4a/1eebe/cd5/7d7ef3f69d479716.webp?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MTUyOTA5Mw==,size_16,color_FFFFFF,t_70)
感谢大佬博主文章:传送门
![在这里插入图片描述](https://img.php1.cn/3cd4a/1eebe/cd5/8373b1277127c518.webp?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MTUyOTA5Mw==,size_16,color_FFFFFF,t_70)