PyQt5将内存中图像数组转换QPixmap、QImage的方法

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