python opencv图像增强（模仿曝光环境）

import cv2
import numpy as np

#读取图片
img = cv2.imread(img_path)
#将bgr转化为hsv
img = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
img = img.astype(np.float)
#获取v通道（颜色亮度通道），并做渐变性的增强
img[:, :, 2] = np.where(img[:, :, 2] > 100, img[:, :, 2] + 20.0, img[:, :, 2])
img[:, :, 2] = np.where(img[:, :, 2] > 150, img[:, :, 2] + 30.0, img[:, :, 2])
img[:, :, 2] = np.where(img[:, :, 2] > 180, img[:, :, 2] + 40.0, img[:, :, 2])
#令大于255的像素值等于255（防止溢出）
img = np.where(img>255, 255, img)
img = img.astype(np.uint8)
res = cv2.cvtColor(img, cv2.COLOR_HSV2BGR)

cv2.imshow('res',res)
cv2.waitKey()

有问题评论问博主