OpenCV——增加视频图像的曝光亮度

# Bright

import cv2
import numpy as np

# 视频路径和输出路径
input_video_path = r'D:\desk\20240713_test\ORI_VIDEOS\rgb_normal_20240713.mp4'
output_video_path = r'D:\desk\20240713_test\ORI_VIDEOS\rgb_bright_1.4_20240713.mp4'

# 打开视频文件
cap = cv2.VideoCapture(input_video_path)

# 获取视频的帧率和帧大小
frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
frame_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
fps = cap.get(cv2.CAP_PROP_FPS)

# 定义输出视频的编码和创建VideoWriter对象
fourcc = cv2.VideoWriter_fourcc(*'XVID')  # 定义编码器和文件格式，XVID是DivX兼容的MPEG-4编码器
out = cv2.VideoWriter(output_video_path, fourcc, fps, (frame_width, frame_height), True)

# 亮度增强的参数
brightness_factor = 50  # 增加的亮度值，根据需要调整

while cap.isOpened():
    ret, frame = cap.read()
    if not ret:
        break

    # 增加亮度
    # 确保像素值在0-255范围内
    # enhanced_frame = cv2.add(frame, brightness_factor)

    enhanced_frame = cv2.convertScaleAbs(frame, alpha=1.2, beta=0)
    # 写入处理后的帧到输出视频
    out.write(enhanced_frame)

    # 显示原始和增强后的帧
    cv2.imshow('Original', frame)
    cv2.imshow('Enhanced', enhanced_frame)

    # 按 'q' 退出
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

# 释放资源
cap.release()
out.release()
cv2.destroyAllWindows()