目录
简述
1. 原理说明
1.2 快速行进方法（INPAINT_TELEA）
2. 实现步骤
2.1 输入图像和掩膜（Mask）
2.2 调用cv2.inpaint()函数
2.3 完整代码示例
2.4 运行结果
3. 掩膜制作方法
4. 差异
4.1 修复效果
4.2 计算效率
5. 总结
简述
在 OpenCV 中，图像修复是一种用于去除图像中不需要的部分（如划痕、污渍等）的技术。OpenCV 提供了两种主要的图像修复算法：基于流体动力学的图像修复方法（Navier-Stokes，NS）和基于快速行进算法（Fast Marching Method，FMM）的修复方法。
1. 原理说明
cv2.inpaint() 基于两种算法：
1.1 Navier-Stokes方法（INPAINT_NS）
- 通过流体动力学模拟，沿等照度线（颜色梯度方向）传播信息。
- 适合修复自然纹理，但计算速度较慢。
1.2 快速行进方法（INPAINT_TELEA）
- 基于像素邻域的加权平均，优先修复边界附近的区域。
- 速度更快，适合实时处理。
2. 实现步骤
2.1 输入图像和掩膜（Mask）
- 原始图像：待修复的图片（需为uint8格式）。
- 掩膜图像：标记需修复区域的二值图像（白色区域为需修复部分）。
2.2 调用cv2.inpaint()函数
inpainted_img = cv2.inpaint(src, mask, inpaintRadius, flags)- src：输入图像（BGR格式）。
- mask：掩膜图像（单通道，非零像素表示需修复区域）。
- inpaintRadius：修复半径（影响周围像素的范围，通常设为3-5）。
- flags：选择算法（cv2.INPAINT_NS 或 cv2.INPAINT_TELEA）。
2.3 完整代码示例
import cv2
import numpy as np
# 读取原始图像和掩膜
image = cv2.imread("D:\\resource\\opencv\\lena_damaged1.png")
# 掩膜需为单通道
mask = cv2.imread("D:\\resource\\opencv\\mask1.png", cv2.IMREAD_GRAYSCALE)
# 检查图像是否读取成功
if image is None or mask is None:
print("Error: image or mask is none !")
exit()
# 图像修复
inpainted_telea = cv2.inpaint(image, mask, inpaintRadius=3, flags=cv2.INPAINT_TELEA)
# inpainted_ns = cv2.inpaint(image, mask, inpaintRadius=3, flags=cv2.INPAINT_NS)
# 显示结果
cv2.imshow("image", image)
cv2.imshow("mask", mask)
cv2.imshow("Inpainted (TELEA)", inpainted_telea)
# cv2.imshow("Inpainted (NS)", inpainted_ns)
cv2.waitKey(0)
cv2.destroyAllWindows()2.4 运行结果
原图：
mask：
TELEA：
注意事项：
-
掩膜必须为单通道二值图像（0表示正常区域，非0表示需修复区域）。
-
修复区域周围需有足够的信息供算法参考，否则效果可能不理想。
-
对于复杂场景（如大面积修复），建议结合深度学习模型（如DeepFill、GAN）。
3. 掩膜制作方法
若没有掩膜图像，可通过以下方式生成：
- 手动标注（如用画图工具标记需修复区域为白色）。
- 程序生成（例如通过阈值分割或边缘检测）：
# 示例：通过颜色阈值生成掩膜（假设修复红色区域）
hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
lower_red = np.array([0, 50, 50])
upper_red = np.array([10, 255, 255])
mask = cv2.inRange(hsv, lower_red, upper_red)4. 差异
4.1 修复效果
cv2.INPAINT_TELEA：
- 修复后的图像通常具有较好的平滑性，在处理小面积的噪声、划痕等损坏时，能够快速生成视觉上较为自然的结果。它对于简单的图像修复任务表现出色，能够有效地去除小瑕疵，使图像看起来更加干净。
- 然而，在处理大面积的缺失区域或具有复杂结构的图像时，可能会出现模糊或丢失细节的问题，因为它更侧重于平滑性，可能会过度平滑图像中的一些重要特征。
cv2.INPAINT_NS：
- 该算法在保留图像的结构信息方面表现更优，尤其适用于修复包含重要线条、纹理或边缘的图像。它能够更好地保持图像的原有结构，使得修复后的区域与周围环境更加融合。
- 但在某些情况下，修复结果可能会显得不够平滑，特别是在处理一些颜色变化较为剧烈的区域时，可能会出现一些轻微的块状效应。
4.2 计算效率
cv2.INPAINT_TELEA：
- 由于其基于快速行进算法，计算速度相对较快。它在处理小规模的修复任务时能够迅速给出结果，对于实时性要求较高的应用场景更为适用。
cv2.INPAINT_NS：
- 该算法需要求解较为复杂的偏微分方程，计算量相对较大，因此处理速度较慢。在处理大尺寸图像或需要修复大面积区域时，可能会花费较多的时间。
5. 总结
在实际应用中选择使用 cv2.INPAINT_TELEA 还是 cv2.INPAINT_NS 可以从以下几个方面进行考虑：
| 考虑因素 | 选择 cv2.INPAINT_TELEA 的情况 | 选择 cv2.INPAINT_NS 的情况 |
|---|---|---|
| 修复区域特征 | 小面积、简单瑕疵修复，如轻微划痕、小斑点、孤立噪声点 | 大面积、复杂结构修复，图像含明显线条、纹理、边缘 |
| 对图像细节和结构的要求 | 追求平滑效果，不太在意细微结构变化 | 注重细节保留，图像细节信息重要 |
| 计算资源和时间限制 | 实时性要求高的场景，如实时视频流修复 | 计算资源充足且不考虑时间成本，如珍贵文物图像离线修复 |
![[css] 黑白主题切换](https://i-blog.csdnimg.cn/direct/b7a920dac69c447ebf1c1568879edfae.png)
