opencv 消除红眼 python 代码

在OpenCV（Python版本）中消除照片中的红眼现象通常涉及到调整图片中的高光部分，因为红眼通常是由于闪光灯照射到眼睛的反光造成的。下面是一个简单的例子，展示了如何使用OpenCV和numpy库来处理这个问题：

import cv2
import numpy as np

def remove_red_eye(image):
    # 将图像从BGR转换为HSV颜色空间
    hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)

    # 定义红眼相关的HSV范围（这里是一个通用范围，可能会根据实际情况调整）
    lower_red = np.array([0, 50, 70])
    upper_red = np.array([10, 255, 255])

    # 创建一个红色通道的掩码
    mask_red = cv2.inRange(hsv, lower_red, upper_red)
    
    # 对原始图像做位运算，去除红眼区域
    red_eye_mask = cv2.bitwise_and(image, image, mask=mask_red)

    # 使用中间值法填充红眼区域
    kernel = np.ones((5, 5), np.uint8)  # 5x5平滑核
    avg_color = cv2.mean(red_eye_mask, (5, 5))
    red_eye_mask = cv2.dilate(mask_red, kernel, iterations=1)
    red_eye_mask_inv = cv2.bitwise_not(red_eye_mask)
    result = cv2.bitwise_or(img=red_eye_mask_inv, img=avg_color, mask=red_eye_mask_inv)

    return result

# 加载图像
image = cv2.imread('input.jpg')
result = remove_red_eye(image)
cv2.imshow('Original Image', image)
cv2.imshow('Red Eye Removed', result)
cv2.waitKey(0)
cv2.destroyAllWindows()