在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图像的格式

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐