使用python进行渐晕校正

import cv2
import numpy as np

def vignette_correction(image):
    # 获取图像的尺寸
    height, width = image.shape[:2]

    # 计算图像中心点坐标
    center_x = width // 2
    center_y = height // 2

    # 计算图像中心到各个像素点的距离
    distance = np.sqrt((np.arange(height)[:, np.newaxis] - center_y) ** 2 +
                       (np.arange(width) - center_x) ** 2)

    # 计算渐晕校正系数
    max_distance = np.sqrt(center_x ** 2 + center_y ** 2)
    correction_factor = distance / max_distance

    # 对图像进行渐晕校正
    corrected_image = image * correction_factor[:, :, np.newaxis]

    return corrected_image

# 读取图像
image = cv2.imread('input.jpg')

# 进行渐晕校正
corrected_image = vignette_correction(image)

# 显示原始图像和校正后的图像
cv2.imshow('Original Image', image)
cv2.imshow('Corrected Image', corrected_image)
cv2.waitKey(0)
cv2.destroyAllWindows()