OpenCV，更改为图像中最接近的颜色

import cv2
import numpy as np

def find_closest_color(image, target_color):
    # 将目标颜色转换为BGR格式
    target_color_bgr = np.array(target_color[::-1], dtype=np.uint8)
    
    # 遍历图像中的每个像素
    for y in range(image.shape[0]):
        for x in range(image.shape[1]):
            pixel_color = image[y, x]
            # 计算当前像素颜色与目标颜色的欧氏距离
            distance = np.linalg.norm(pixel_color - target_color_bgr)
            if distance < min_distance:
                min_distance = distance
                closest_color = pixel_color
    
    return closest_color

def change_color_in_image(image_path, target_color):
    # 读取图像
    image = cv2.imread(image_path)
    
    # 找到最接近的颜色
    closest_color = find_closest_color(image, target_color)
    
    # 将图像中的所有像素更改为最接近的颜色
    for y in range(image.shape[0]):
        for x in range(image.shape[1]):
            image[y, x] = closest_color
    
    # 保存修改后的图像
    cv2.imwrite('modified_image.jpg', image)

# 示例用法
image_path = 'example.jpg'
target_color = [255, 0, 0]  # 红色
change_color_in_image(image_path, target_color)