文章目录
- 1. 格式转换
- 2. 图像操作
1. 格式转换
参考python图像的基本操作
在深度学习算法中,会涉及到如下与图像处理相关的操作和转换。基本的库函数为:opencv,pillow,torchvision(tensor),numpy。
pillow->tensor
tensor->pillow
第一,图像读取使用PIL,直接使用torchvision函数将PIL图像转换为tensor。
第二,使用torchvision函数转换从tensor转换为PIL图像,使用PIL进行图像存储。
from PIL import Image
from torchvision import transforms
img = transforms.ToTensor()(img) # pillow->tensor, 自动归一化0-1
img_pil = transforms.ToPILImage()(img) # tensor->pillow格式,自动转换为0-255
opencv->tensor
img = cv.imread('image/0