本文主要是介绍opencv-python实战项目十二:去除鼠标选择区域的水印,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
文章目录
- 一,简介
- 二,算法实现:
- 2.1 鼠标事件定位水印区域
- 2.2 水印区域修复:
- 三,整体代码:
- 四,效果:
一,简介
在日常生活中,我们常常会遇到带有水印的图片,这些水印虽然保护了版权,但有时也会影响图像的美观和使用。本次博客将向您介绍如何运用OpenCV这一强大的计算机视觉库,轻松实现图像去水印,让您的照片恢复纯净之美。在这篇文章中,我们将深入探讨OpenCV在图像处理领域的应用,特别是如何通过简单的代码操作,定位并去除图像中的水印。
二,算法实现:
2.1 鼠标事件定位水印区域
在本次项目中,我们使用OpenCV库来读取用户的鼠标状态,以便选择需要去除水印的区域。首先,通过初始化全局变量来监控鼠标事件和矩形绘制状态。当用户按下鼠标左键时,算法会开始绘制矩形,并记录下起始点坐标(ix, iy)。在鼠标移动过程中,如果绘制状态为真,则会在原始图像的副本上动态绘制一个绿色矩形框,其边界由起始点和当前鼠标位置共同确定。用户释放鼠标左键后,绘制状态会被关闭,并记录矩形的终点坐标(fx, fy)。同时,在掩码图像上绘制一个与之坐标相同的白色填充矩形。这个掩码图像将在后续的水印去除操作中使用。
代码:
def draw_rectangle(event, x, y, flags, param):global ix, iy, fx, fy, drawing, maskif event == cv2.EVENT_LBUTTONDOWN:drawing = Trueix, iy = x, yelif event == cv2.EVENT_MOUSEMOVE:if drawing:img_draw = img.copy()cv2.rectangle(img_draw, (ix, iy), (x, y), (0, 255, 0), 2)cv2.imshow('image', img_draw)elif event == cv2.EVENT_LBUTTONUP:drawing = Falsefx, fy = x, ycv2.rectangle(mask, (ix, iy), (fx, fy), (255,255,255), -1)
2.2 水印区域修复:
本次水印区域修复运用的是opencv自带函数cv2.inpaint()
cv2.inpaint()是OpenCV库中的一个函数,用于修复图像中的损坏区域。它通过分析图像中未损坏的像素,来推断并填充损坏区域的像素值,从而使得修复后的区域与周围图像内容尽可能融合,减少视觉上的不连续性。
函数介绍:
输入:
● src:输入的原始图像,通常是8位单通道或3通道(彩色)图像。
● mask:一个8位单通道图像,与src具有相同的尺寸。掩码图像中的非零像素表示需要修复的损坏区域。
● inpaintRadius:一个正数,表示修复过程中考虑的邻域半径(以像素为单位)。这个半径定义了在修复一个像素时,函数将考虑多少周围的像素。
● flags:修复方法标志,有两个选项:
○ cv2.INPAINT_TELEA:使用基于快速行进方法的算法,由Alexandru Telea提出。
○ cv2.INPAINT_NS:使用基于Navier-Stokes方程和流体动力学的方法。
返回值:
返回一个图像,它与src具有相同的尺寸和类型,其中损坏的区域已经被修复。
cv2.inpaint()函数常用于去除图像中的小缺陷,如噪点、划痕或小面积的水印等。不过,其修复效果取决于损坏区域的大小和图像的纹理复杂度,有时可能需要调整inpaintRadius和flags参数来获得最佳效果。
代码:
inpainted_img = cv2.inpaint(img, mask, 3, cv2.INPAINT_TELEA)
三,整体代码:
import cv2
import numpy as np# 初始化全局变量
drawing = False
ix, iy = -1, -1
fx, fy = -1, -1# 鼠标回调函数
def draw_rectangle(event, x, y, flags, param):global ix, iy, fx, fy, drawing, maskif event == cv2.EVENT_LBUTTONDOWN:drawing = Trueix, iy = x, yelif event == cv2.EVENT_MOUSEMOVE:if drawing:img_draw = img.copy()cv2.rectangle(img_draw, (ix, iy), (x, y), (0, 255, 0), 2)cv2.imshow('image', img_draw)elif event == cv2.EVENT_LBUTTONUP:drawing = Falsefx, fy = x, ycv2.rectangle(mask, (ix, iy), (fx, fy), (255,255,255), -1)# 加载图像
img = cv2.imread(r'F:\traditional_vison\1a0c30580ad44357a4d625845ec32db5.png')
img = cv2.resize(img,(0,0),fx=0.5,fy=0.5)
mask = np.zeros(img.shape[:2], dtype=np.uint8)
cv2.namedWindow('image')
cv2.setMouseCallback('image', draw_rectangle)while True:cv2.imshow('image', img)k = cv2.waitKey(1) & 0xFFif k == ord('q'):break# 检查是否已经框选了一个区域
if ix >= 0 and iy >= 0 and fx >= 0 and fy >= 0:# 使用 inpaint 函数修复图像inpainted_img = cv2.inpaint(img, mask, 3, cv2.INPAINT_TELEA)cv2.imshow('Inpainted Image', inpainted_img)cv2.waitKey(0)
cv2.destroyAllWindows()
四,效果:
去除水印前:
去除水印后:
这篇关于opencv-python实战项目十二:去除鼠标选择区域的水印的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!