利用 `OpenCV` 和 `Matplotlib` 库进行图像读取、颜色空间转换、掩膜创建、颜色替换
这段代码主要涉及图像处理任务，利用 OpenCV 和 Matplotlib 库进行图像读取、颜色空间转换、掩膜创建、颜色替换等操作。下面是对代码的逐部分解释和归类：
1. 导入库
import cv2
import matplotlib.pyplot as plt
import numpy as np
cv2: OpenCV库，用于图像处理。matplotlib.pyplot: 用于绘制和显示图像。numpy: 用于处理数组，尤其是在图像数据处理中的数值操作。
2. 读取和转换图像颜色空间
image = cv2.imread('E:/insulator/1/1/picture/lion/4.png')
image = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
cv2.imread(): 读取图像。cv2.cvtColor(): 将图像从 BGR 颜色空间转换为 HSV 颜色空间（色调、饱和度、明度），这有助于根据颜色值进行更准确的处理。
3. 创建颜色掩膜
lower_color = np.array([127, 29, 29]) # 红色低范围
upper_color = np.array([255, 255, 246]) # 红色高范围
mask = cv2.inRange(image, lower_color, upper_color)
- 使用
cv2.inRange()创建一个掩膜，只保留在lower_color和upper_color范围内的颜色（即红色范围）。
4. 应用掩膜并更改颜色
new_color = np.array([156, 163, 175], dtype=np.uint8)
background = np.array([255, 255, 255], dtype=np.uint8)
colored_image = np.zeros_like(image)
colored_image[mask > 0] = new_color
colored_image[mask == 0] = background
- 创建一个新的空白图像
colored_image，然后根据掩膜将红色区域替换为new_color，背景区域设置为白色。
5. 保存结果
cv2.imwrite('E:/insulator/1/1/picture/lion/colored_image.png', colored_image)
- 使用
cv2.imwrite()将处理后的图像保存为colored_image.png。
6. 操作与图像分析
white_mask = cv2.inRange(image1, (204, 40, 52), (229, 5, 8))
image1[white_mask] = (0, 150, 105)
- 创建白色区域的掩膜，将掩膜区域的颜色更改为指定的绿色
(0, 150, 105)。
7. 处理图像中的像素值
image1[image1 < 255] = 25
image1[image1 > 11] = (255, 0, 0)
- 通过条件操作修改像素值：对于像素值小于 255 的区域，将其设为 25；对于像素值大于 11 的区域，将其设为蓝色
(255, 0, 0)。
8. 平均化图像（通道降维）
image2 = image2.mean(axis=2)
- 将图像
image2的通道降维，转为灰度图（通过取平均值）。
9. 根据像素值进行条件替换
image1[np.logical_and(image3 > 222, image2 > 0)] = (0, 255, 0)
- 使用
np.logical_and()结合多个条件判断，修改符合条件的像素的颜色。
10. 更多条件操作与图像显示
image1[np.logical_and(image3 > 222, image2 > 0)] = (0, 255, 0)
image[np.logical_and(image1 > 46, image1 < 88)] = (255, 0, 0)
- 根据多个条件修改图像的像素值，修改符合条件的像素为指定颜色。
11. 噪声添加与二值化
noise = np.random.rand(*image.shape)
thres = 0.3
image[noise > thres] = 255
image[image < 240] = 0
image[image >= 240] = 1
- 使用随机噪声
np.random.rand()修改图像的像素值，并根据阈值将图像转为二值图像（0 或 1）。
12. 根据掩膜对新图像进行处理
jueyuanzi_mask = cv2.imread("./jueyuanzi_mask.png")
new_image[np.logical_and(jueyuanzi_mask != 255, image != 1)] = (185, 28, 28)
chuanshuxian_mask = cv2.imread("./chuanshuxian_mask.png")
new_image[np.logical_and(chuanshuxian_mask != 255, image != 1)] = (255, 255, 255)
- 读取掩膜图像（
jueyuanzi_mask和chuanshuxian_mask），根据掩膜的条件修改图像new_image的颜色。
13. 保存最终图像
new_image = cv2.cvtColor(new_image, cv2.COLOR_RGB2BGR)
cv2.imwrite("new_image.png", new_image)
- 将最终图像
new_image转换为 BGR 颜色空间并保存为new_image.png。
总结
这段代码涉及图像处理中的几个常见任务：
- 颜色空间转换：将图像从 BGR 转换为 HSV。
- 掩膜创建与颜色替换：通过颜色范围提取特定区域，并替换为新颜色。
- 图像的条件修改：根据像素值或掩膜条件改变图像区域的颜色。
- 噪声添加与图像二值化：模拟噪声并进行图像二值化处理。
- 图像合成与保存：通过多个操作合成新的图像，并最终保存。
这些操作可用于目标检测、图像分割、区域标记等应用，尤其在图像处理中用于特定颜色的提取与修改。