OpenCV/CV2,PIL,Tensor之间如何互相转化

参考资料:

torchvision文档

众所周知OpenCV/CV2和PIL是Python领域最常使用的图像处理库,而Pytorch Tensor则是我们在深度学习领域接触最多的张量数据结构,知道这几种格式之间的相互转化方式是非常有必要的。三类,一共6种相互转化的链路,下面一一道来:

1. OpenCV/CV2(ndarray)

cv2不像PIL Image那样有特定的图像存储类,因此cv2格式其实是一个伪命题。使用cv2读取图片,实际上获得的是一个ndarray。ndarray更是我们的老熟人,它是numpy包中的张量数据类型。

ndarray to tensor:

image_nd = cv2.imread("XX.jpg")
A = ( torch.tensor(image_nd) / 255. ).flip(-1)

上面是一个典型的代码片,需要注意cv2读到的ndarray是uint8类型,并且默认通道顺序是BGR,因此需要先除以255然后反转最后一个维度(Pytorch不支持[::-1]这种切片逆转语法)。

ndarray to pil

image_nd = cv2.imread("/data1/chaishang.cs/Tryon/31_human.jpg")
image_pil = Image.fromarray(image_nd[..., ::-1])

只需使用Image.fromarray方法,需要注意pil图片的通道顺序是RGB,因此也需要先对通道进行反转。

**  2. PIL image**

**  pil to tensor**

import torchvision.transforms.functional as F
F.pil_to_tensor(image_pil) / 255.

pil to ndarray

image_nd = np.array(image_pil)

3. Tensor

**  tensor to ndarray**

直接对tensor调用.numpy()函数即可,如果tensor位于cuda上,需要先运行.cpu()

tensor to pil

import torchvision.transforms.functional as F
F.to_pil_image(tensor)

需要注意tensor需要是[C, H, W]的格式,并且归一化到[0, 1]区间。

相关推荐
管二狗赶快去工作!5 分钟前
体系结构论文(五十四):Reliability-Aware Runahead 【22‘ HPCA】
人工智能·神经网络·dnn·体系结构·实时系统
AI绘画君14 分钟前
Stable Diffusion绘画 | AI 图片智能扩充,超越PS扩图的AI扩图功能(附安装包)
人工智能·ai作画·stable diffusion·aigc·ai绘画·ai扩图
AAI机器之心16 分钟前
LLM大模型:开源RAG框架汇总
人工智能·chatgpt·开源·大模型·llm·大语言模型·rag
Evand J37 分钟前
物联网智能设备:未来生活的变革者
人工智能·物联网·智能手机·智能家居·智能手表
HyperAI超神经1 小时前
Meta 首个多模态大模型一键启动!首个多针刺绣数据集上线,含超 30k 张图片
大数据·人工智能·深度学习·机器学习·语言模型·大模型·数据集
sp_fyf_20241 小时前
计算机前沿技术-人工智能算法-大语言模型-最新研究进展-2024-10-03
人工智能·算法·机器学习·计算机视觉·语言模型·自然语言处理
新缸中之脑1 小时前
10个令人惊叹的AI工具
人工智能
学步_技术1 小时前
自动驾驶系列—线控悬架技术:自动驾驶背后的动力学掌控者
人工智能·机器学习·自动驾驶·线控系统·悬挂系统
Eric.Lee20211 小时前
数据集-目标检测系列- 螃蟹 检测数据集 crab >> DataBall
python·深度学习·算法·目标检测·计算机视觉·数据集·螃蟹检测
DogDaoDao2 小时前
【预备理论知识——2】深度学习:线性代数概述
人工智能·深度学习·线性代数