在PyQt5中,一般QPixmap、QImage通常都是直接从硬盘上读取图片获得,但是如果我想将torch.tensor类型转化成QPixmap在QLabel上显示应该怎么做呢?
思路是可以将torch.tensor转为numpy,再将其转为PIL图像,最后利用toqpixmap() 和toqimage() 方法转为QPixmap和QImage
from PIL import Image
import numpy as np
# 假设当前内存中的img类型为 32x32x3 type为tensor.uint8类型
img_pil = Image.fromarray(np.uint8(img.numpy())
img_pix = img_pil.toqpixmap() #QPixmap
img_img = img_pil.toqimage() #QImage
在这里再补充一点图像处理相关的坑点:
使用matplotlib.pyplot显示图片,图片不能是浮点数,必须为整型
PIL图像的维度是[C,H,W],维度为[H,W,C]的数组图像才能用matplotlib画出来,[C,H,W]的数组图像是不能用matplotlib画出来的,除非转化为PIL图像的格式