2018/05/24 日常挖坑填坑
pytorch用Tensor处理图像数据,想显示图片。直接调用
vutils.save_image(rgb.cpu().float(), os.path.join("./color/1.jpg"))
如果是Variable类型,则
rgb.data.cpu().float()
不要用
RGB=transform.ToPILImage()(rgb) plt.imshow(RGB) plt.show() 可能显示的颜色会有问题,导致自己会怀疑程序某些地方写错了。花了近3个小时才找到这个bug,特地记录。