如何将OpenCV格式的数组形式的图片转换为PIL格式数组并进行transforms图片数据增强
时间: 2024-05-09 09:21:26 浏览: 75
从npz数据中提取成jpg图片格式、并且计算mask的前景值以及划分数据集
可以使用以下代码将OpenCV格式的数组形式的图片转换为PIL格式数组,并进行transforms图片数据增强:
```
import cv2
import numpy as np
from PIL import Image
from torchvision import transforms
# 读取OpenCV格式的图片
img_cv = cv2.imread('image.jpg')
# 将OpenCV格式的图片转换为PIL格式数组
img_pil = Image.fromarray(cv2.cvtColor(img_cv, cv2.COLOR_BGR2RGB))
# transforms图片数据增强
transform = transforms.Compose([
transforms.Resize(size=(224, 224)),
transforms.ToTensor(),
transforms.Normalize(mean=[0.485, 0.456, 0.406],
std=[0.229, 0.224, 0.225])
])
img_tensor = transform(img_pil)
# 输出PIL格式数组并进行transforms图片数据增强后的图片
print(img_tensor)
```
阅读全文