python对视频调色 写一份代码

Python 对视频调色通常涉及处理图像数据，可以利用 OpenCV (Open Source Computer Vision Library) 等库来操作视频帧。下面是一个简单的例子，展示如何使用 OpenCV 进行基本的色彩调整，如亮度、对比度和饱和度：

import cv2
import numpy as np

# 加载视频
cap = cv2.VideoCapture('your_video.mp4')

# 创建一个视频 writer 对象用于保存处理后的视频
fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # 使用常见的编码
out = cv2.VideoWriter('output.mp4', fourcc, cap.get(cv2.CAP_PROP_FPS), (int(cap.get(cv2.CAP_PROP_FRAME_WIDTH)), int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))))

while True:
    # 读取一帧
    ret, frame = cap.read()
    
    # 如果帧读取成功
    if ret:
        # 调整颜色参数（这里只是简单示例）
        # 亮度调整
        brightness = 0.5  # 0.5 表示增加亮度
        adjusted_frame = cv2.add(frame, brightness * np.array([[[0, 0, 0]]], dtype=np.uint8))

        # 对比度和饱和度调整
        contrast = 1.2  # 提高对比度
        saturation = 1.3  # 提高饱和度
        hsv = cv2.cvtColor(adjusted_frame, cv2.COLOR_BGR2HSV)
        h, s, v = cv2.split(hsv)
        s = cv2.multiply(s, saturation)
        combined = cv2.merge((h, s, v))
        adjusted_frame = cv2.cvtColor(combined, cv2.COLOR_HSV2BGR)

        # 写入处理后的帧
        out.write(adjusted_frame)

        # 显示原始和处理后的画面，这里仅作查看，实际应用可能不需要显示
        cv2.imshow('Original Frame', frame)
        cv2.imshow('Processed Frame', adjusted_frame)
        
        # 按 'q' 键退出循环
        if cv2.waitKey(1) & 0xFF == ord('q'):
            break

    else:
        break

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