彩色图像中阴影的去除

import cv2
import numpy as np

def remove_shadow(image):
    # 转换为HSV颜色空间
    hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
    
    # 分离通道
    h, s, v = cv2.split(hsv)
    
    # 应用自适应阈值处理
    _, thresh = cv2.threshold(v, 50, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)
    
    # 形态学操作去除噪声
    kernel = np.ones((5, 5), np.uint8)
    opening = cv2.morphologyEx(thresh, cv2.MORPH_OPEN, kernel, iterations=2)
    
    # 创建遮罩
    mask = cv2.bitwise_not(opening)
    
    # 应用遮罩到原始图像
    result = cv2.bitwise_and(image, image, mask=mask)
    
    return result

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

# 去除阴影
result_image = remove_shadow(image)

# 显示结果
cv2.imshow('Original Image', image)
cv2.imshow('Shadow Removed Image', result_image)
cv2.waitKey(0)
cv2.destroyAllWindows()