使用OpenCV去除图片水印的完整指南

使用OpenCV去除图片水印的完整指南

引用

CSDN

1.

https://blog.csdn.net/qq_67037885/article/details/138246692

在图像处理领域，去除图片上的水印是一个常见的需求场景。本文将介绍如何使用OpenCV库中的inpaint函数来去除图片上的水印，通过具体的代码示例和效果展示，帮助读者掌握这一实用技能。

使用到的API

OpenCV提供了一个专门用于去除水印的函数inpaint，其函数原型如下：

inpaint(
    img,  // 需要恢复的图像
    mask,    // 标记需要去除水印的部分，要求是黑底白色单通道图像
    inpaintRadius,   // 修复半径：破损位置像素点的修复半径
    flags               // 方法：INPAINT_NS（Navier-Stokes-based method）或INPAINT_TELEA（Telea's method）
)

准备工作

在进行水印去除之前，需要准备两张图片：

1. img：原始图片，包含需要去除的水印
2. mask：掩码图片，用于标记水印的位置

代码实现

下面是一个完整的Python代码示例，展示了如何使用OpenCV去除图片上的水印：

import cv2
import numpy as np

# 读入原始图片
roi = cv2.imread('roi.jpg')
cv2.imshow("roi", roi)

# 读入掩码图片
mask = cv2.imread('watermark.jpg', 0)
mask = cv2.resize(mask, (roi.shape[1], roi.shape[0]))

# 对掩码图片进行二值化处理
ret, threshold = cv2.threshold(mask, 0, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)
cv2.imshow("mask", threshold)

# 进行水印去除
dst = cv2.inpaint(roi, threshold, 5, cv2.INPAINT_NS)

# 显示最终结果
cv2.imshow('dst', dst)
cv2.waitKey(0)

效果展示

去除水印前后的对比效果如下图所示：

通过上述步骤，可以有效地去除图片上的水印。需要注意的是，水印去除的效果会受到原始图片质量、水印类型以及掩码图片准确度的影响，因此在实际应用中可能需要根据具体情况进行调整。