PIL和opencv读取图片

import os.path
from PIL import Image
import cv2
import numpy as np
import matplotlib.pyplot as plt
import torchvision.transforms as transforms##Image 读取图片
image = Image.open("1.jpg").convert("RGB")
##opencv 读取图片
image2 = cv2.imread("1.jpg")
print("Image read",image.size)  # w h
print("Opencv read",image2.shape)  # h w c

PIL 读取的图片是PIL格式，需要转换为numpy或者tensor格式进行数据处理。opencv读取的图片直接就是numpy格式。

PIL与tensor的相互转换

可以利用torchvision中的transforms类中ToTensor()和ToPILImage()两个函数将PIL的格式和tensor格式互相转换。注意，转换后的像素点发生的变化..。

这里opencv像素点范围是0-255，tensor像素点范围是0-1

import os.path
from PIL import Image
import cv2
import numpy as np
import matplotlib.pyplot as plt
import torchvision.transforms as transformsimage = Image.open("1.jpg").convert("RGB")
image2 = cv2.imread("1.jpg")
image_tensor = transforms.ToTensor()(image)
image_recover = transforms.ToPILImage()(image_tensor)
image_tensor_add = image_tensor.unsqueeze(0)  ##增加一维
print("Image read",image.size)  # w h
print("Opencv read",image2.shape)  # h w c
print("tensor",image_tensor.size()) #c h w
print("tensor recover to Image",image_recover.size)
print("tensor add ",image_tensor_add.size())

numpy 和 tensor 相互转换

注意：numpy格式的图片是numpy.uint8，tensor格式的图片的数据类型是torch.float32。

import os.path
from PIL import Image
import cv2
import numpy as np
import matplotlib.pyplot as plt
import torchvision.transforms as transforms#方法一：直接转换
image_tensor_from_numpy = transforms.ToTensor()(image)
print("image_tensor_from_numpy： ",image_tensor_from_numpy.shape)
#方法二
image_tensor_from_numpy2 = cv2.cvtColor(image,cv2.COLOR_BGR2RGB) #先将 BGR 变为 RGB
image_tensor_from_numpy2 = torch.from_numpy(image_tensor_from_numpy2.transpose((2, 0, 1)))
# 将 (h,w,c)转换为 (c,h,w)
image_tensor_from_numpy2 = image_tensor_from_numpy2.float().div(255) #归一化到[0,1]
print("image_tensor_from_numpy2 :",image_tensor_from_numpy2.size())

image_numpy_form_Image = np.array(image)
image_numpy_form_Image = cv2.cvtColor(image_numpy_form_Image,cv2.COLOR_RGB2BGR)
print("image_numpy_form_Image ",image_numpy_form_Image.shape)

注意：PIL和opencv读取的图片格式不同，因此通常进行格式转换

参考文献：

pytorch中图片类型的转换——PIL、tensor、numpy_长歌丶采薇的博客-CSDN博客_pil转numpy

Pytorch中Tensor与各种图像格式（PIL 和 Numpy）的相互转化_Phoenixtree_DongZhao的博客-CSDN博客_pil 转化为numpy

numpy PIL tensor之间的相互转换相关推荐

[开发技巧]·TensorFlow中numpy与tensor数据相互转化（支持tf1.x-tf2.x）
推荐对比阅读: [开发技巧]·PyTorch中Numpy,Tensor与Variable深入理解与转换技巧『带你学AI』带你学AI与TensorFlow2实战之入门初探:如何速成深度学习开发鉴于t ...
c# image转换为bitmap_Python PIL.Image与numpy.array之间的相互转换
前言有时我们使用PIL库读入图像数据后需要查看图像数据的维度,比如shape,或者有时我们需要对图像数据进行numpy类型的处理,所以涉及到相互转化,这里简单记录一下. 方法当使用PIL.Imag ...
PyTorch | Tensor、Numpy、Variable之间的转换
其中,Numpy和Variable之间的转换可以通过tensor推导出来. Numpy->Variable:拆分为两步, Numpy->Tensor: torch.from_numpy(t ...
一文看懂所有图像与numpy矩阵（三维、四维均包含）之间的相互转换
图像与numpy矩阵之间的相互转换: 1.RGB–>灰度图: from PIL import Image # 加载RGB图片 I = Image.open('/home/timg.jpg') I ...
PIL，cv2读取类型及转换，以及PIL，numpy，tensor格式以及cuda，cpu的格式转换
PIL,cv2读取类型,以及PIL,numpy,tensor格式以及cuda,cpu的格式转换一.PIL,cv2读取数据图片以及之间的转换二.PIL,数组类型以及tensor类型的转换 1.PIL ...
list,numpy,tensor之间相互转换的方法
list,numpy,tensor之间相互转换的方法: a=[[1,2],[3,4]]#list print(a) b=np.array(a)#list->numpy print(b) c=to ...
2021-07-26 TypeError: can‘t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the
can't convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first. 意思是:如果 ...
DLPack构建跨框架深度学习编译器-实现张量之间的相互转换【pytorch】
文章目录 1 概述 2 pytorch使用torch.utils.dlpack实现DLPack与张量之间的相互转换 2.1 将DLPack解码为张量. 2.2 将张量转换为DLPack 3 在 PyT ...
PInvoke复习之深入理解char*与wchar_t*与string以及wstring之间的相互转换
本篇文章是对c++中的char*与wchar_t*与string以及wstring之间的相互转换进行了详细的分析介绍,需要的朋友参考下 #ifndef USE_H_ #define USE_H_ #i ...

numpy PIL tensor之间的相互转换

PIL和opencv读取图片

PIL与tensor的相互转换

numpy 和 tensor 相互转换

numpy PIL tensor之间的相互转换相关推荐

最新文章

热门文章