python中PIL.Image和opencv处理图像方法及区别

PIL.Image

# 读取图片
img = Image.open(img_path)

# 读入后矩阵维度
(h, w, c)

# 交换维度（通常在送入torch跑模型之前要转换）
# 表示变换后的第0维用原来的第2维，第1维用原来的第0维，第2维用原来的第1维
img.transpose(2, 0, 1)

# 获取图片大小（属性）
img.size

# 保存图片
img.save(save_path)

# 图片转为RGB
img.convert('RGB')

---

cv2

# 读取图片
img = cv2.imread(img_path)

# 保存图片
img = cv2.imwrite(save_path, img)

# 显示图片
img = cv2.imshow("window name",img)

# 获取图片维度
img.shape

# BGR转RGB
img = cvtColor(img, COLOR_BGR2RGB)

# 画框（参数依次为：图片array,左上角坐标，右下角坐标，框颜色，线粗细）
cv2.rectangle(img, (240, 0), (480, 375), (0, 255, 0), 2)

---

区别

• PIL.Image读入的图片默认顺序是RGB，opencv读入的图片默认顺序是BGR
• opencv读入的图片是以ndarray形式保存，img则是特定Image形式保存

---

转换

# opencv转Image
img = cv2.imread(img_path)
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
img = Image.fromarray(img)

# Image转opencv
img = Image.open(img_path)
cv2.cvtColor(np.asarray(img), cv2.COLOR_RGB2BGR)