OpenCV：图像修复

目录

简述

1. 原理说明

1.1 Navier-Stokes方法（INPAINT_NS）

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：

注意事项：

1. 掩膜必须为单通道二值图像（0表示正常区域，非0表示需修复区域）。

2. 修复区域周围需有足够的信息供算法参考，否则效果可能不理想。

3. 对于复杂场景（如大面积修复），建议结合深度学习模型（如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 可以从以下几个方面进行考虑：