对数据进行曝光增强：使用Python和OpenCV提升视频亮度

本文链接：https://blog.youkuaiyun.com/very_big_house/article/details/144073947

在视频处理领域，调整视频的曝光度是常见的需求之一。无论是因为拍摄环境的光线不足，还是为了改善视觉效果，对视频进行亮度增强都是一项实用的技术。本文将介绍如何使用Python和OpenCV库对视频进行亮度增强，以提升视频的曝光度。

亮度增强的原理

亮度增强通常涉及到调整图像的HSV（色相、饱和度、值）颜色空间中的“值”（V）通道。在HSV颜色空间中，V通道代表图像的亮度。通过增加V通道的值，我们可以提高图像的亮度，从而使图像看起来更亮。

实现亮度增强的步骤

1. 导入必要的库

首先，我们需要导入Python中处理图像和视频的库——OpenCV，以及用于文件路径操作的os库。

import cv2
import os
import numpy as np

2. 定义亮度增强函数

我们定义一个函数brightnessEnhancement，它接受一个图像帧和亮度增强因子作为参数，返回增强后的图像。

def brightnessEnhancement(frame, brightness_factor):
    hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
    hsv[:,:,2] = np.clip(hsv[:,:,2]*brightness_factor, 0, 255).astype('uint8')
    return cv2.cvtColor(hsv, cv2.COLOR_HSV2BGR)

3. 定义视频处理函数

接下来，我们定义一个函数process_video，它接受输入视频路径和输出视频路径，对视频进行亮度增强，并保存增强后的视频。

def process_video(input_path, output_path):
    cap = cv2.VideoCapture(input_path)
    fps = cap.get(cv2.CAP_PROP_FPS)
    width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
    height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
    out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
    brightness_factor = 1.5  # 亮度增强因子，可以根据需要调整
    while cap.isOpened():
        ret, frame = cap.read()
        if not ret:
            break
        brightened_frame = brightnessEnhancement(frame, brightness_factor)
        out.write(brightened_frame)
    cap.release()
    out.release()

4. 设置视频路径并处理视频

最后，我们设置输入和输出视频的路径，并调用process_video函数来处理视频。

videoDir = "视频路径"
saveVideoDir = "保存路径"
input_path = os.path.join(videoDir, "视频名称")  
output_path = os.path.join(saveVideoDir, "保存的视频名称")
process_video(input_path, output_path)

需要注意的是：亮度增强因子brightness_factor可以根据视频的实际情况进行调整。增强因子为0.0将产生黑色图像，为1.0将保持原始图像。如果设置得过高，可能会导致图像过曝。

最后附完整代码及改进后的图像对比

import cv2
import os
import numpy as np
def brightnessEnhancement(frame, brightness_factor):
    hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
    hsv[:,:,2] = np.clip(hsv[:,:,2]*brightness_factor, 0, 255).astype('uint8')
    return cv2.cvtColor(hsv, cv2.COLOR_HSV2BGR)
def process_video(input_path, output_path):
    cap = cv2.VideoCapture(input_path)
    fps = cap.get(cv2.CAP_PROP_FPS)
    width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
    height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
    out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
    brightness_factor = 1.5
    while cap.isOpened():
        ret, frame = cap.read()
        if not ret:
            break
        brightened_frame = brightnessEnhancement(frame, brightness_factor)
        out.write(brightened_frame)
    cap.release()
    out.release()
videoDir = "视频路径"
saveVideoDir = "保存路径"
input_path = os.path.join(videoDir, "视频名称")  
output_path = os.path.join(saveVideoDir, "保存的视频名称")
process_video(input_path, output_path)

原始图像