python numpy,PIL,opencv 操作总结

base64解码

# 参数支持bytes或str
img_bytes = base64.b64decode('image')

将bytes转为bgr图片

import cv2
import numpy as np
img_bgr = cv2.imdecode(np.asarray(bytearray(img_bytes), dtype="uint8"), cv2.IMREAD_COLOR)

img_bgr：nparray 多维数组

用cv2读取图片并转为bytes

import cv2
img = cv2.imread('a.jpg')
# '.jpg'表示把当前图片img按照jpg格式编码
_, img_encode = cv2.imencode(".jpg", img)
img_bytes = bytes(img_encode)

imread()读取的图片，直接就是numpy.ndarray格式
imencode的img参数一定是一个BGR格式的uint8 ndarray

提取图片上某一像素点的颜色

需要把rgb/bgr颜色模型转成hsv颜色模型，可以更方便优雅地判断颜色
h表示色调；s表示饱和度；v表示明度

import cv2
img_hsv = cv2.cvtColor(img_bgr, cv2.COLOR_BGR2HSV)
h, s, v = img_hsv[location_y, location_x]

提取图片的尺寸

size = img_bgr.shape
width = size[1]  # 宽度
height = size[0]  # 高度

判断图片格式

import imghdr
image_type = imghdr.what('a.jpg', img_bytes)

imghdr.what第一个参数接受文件名；第二个参数接受字节流
如果只提供第一个参数，将会返回对应的图片类型
如果提供了第二个参数，那么第一个参数会被忽略，返回字节流的图片类型
不可以只提供第二个参数，会报错

用PIL读取图片并转为ndarray

from PIL import Image
import numpy as np

img_PIL = Image.open(imagepath)
img = np.array(img_PIL)

img_PIL 是 PIL 类型

增加对比度

PIL模块中的ImageEnhance类，用于图像增强处理，提供增强(减弱)图像的亮度、对比度、色度、锐度等。

from PIL import Image
from io import BytesIO
image = Image.open(BytesIO(img_bytes))
image_enhance = ImageEnhance.Contrast(image) # 返回一个增强对象
contrast = 1.5
image_contrasted = image_enhance.enhance(contrast)
img_byte = BytesIO()
image_contrasted.save(img_byte, format='JPEG')
binary_content = img_byte.getvalue()  # 将Image对象转为bytes

image_contrasted.save()还接受路径参数，可以将转换后的图片保存到对应位置

将 OpenCV 图像保存到指定的文件

cv2.imwrite("../image/b.png",img_bgr)

img_bgr：要保存的 OpenCV 图像，nparray 多维数组

用窗口显示图片

cv2.imshow("title", img)
cv2.waitKey(0)
cv2.destroyAllWindows()

img是numpy.ndarray格式