torch将tensor类型的数据进行可视化

1、将tensor转为array类型。

假设此时image是tensor类型，利用下面语句转为array。

img_np = image.numpy()

2、利用cv2将array转为图片并保存

img_np = img_np[0].transpose([1,2,0])      #取出其中一张并转换维度
img_np = (img_np - np.min(img_np))/(np.max(img_np) - np.min(img_np)) *255.0  #转为0-255
img_np = img_np.astype(int)                #转换数据类型
cv2.imwrite('./1.jpg', img_np)             #保存为图片