Pytorch的一些坑

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，特地记录。