python+opencv实现图片/短视频一键去水印

0 前言

在制作ppt个人文章或者分享图片过程中，经常会遇到一些带有水印的情况，不少人都希望能够去除这些水印，提高图片和视频的观看体验。本文将介绍如何使用Python+opencv实现图片的去水印功能。
在这里插入图片描述
👉 点击链接体验在线demo

1 准备工作

首先，我们需要导入OpenCV和NumPy库。OpenCV是一个开源的计算机视觉库，可以用于图像和视频处理，NumPy是Python的一个科学计算库，提供了高性能的多维数组对象。

import cv2
import numpy as np

2 读取图片或视频

在代码中，我们定义了一个变量pmode来指定处理模式，可以是"image"或"video"。当pmode为"image"时，我们从指定路径读取一张图片；当pmode为"video"时，我们通过cv2.VideoCapture()函数从指定路径读取一个视频。

pmode="image"
path = "316.jpg"   #记得不要有中文路径
vieodpath="350.mp4"
cap=cv2.VideoCapture(vieodpath)if pmode=="video":ret,img = cap.read()
else:img = cv2.imread(path)
height,width = img.shape[0:2]

3 添加回调获取鼠标绘制水印区域

接下来，我们获取图片的高度和宽度，并定义了一些全局变量，用于记录鼠标事件的状态和坐标。同时，我们创建了一个与窗口绑定的回调函数，用于实现鼠标事件的交互操作。

# 鼠标回调函数
def draw_circle(event, x, y, flags, param):global ix, iy, drawing, mode, imgif event == cv2.EVENT_LBUTTONDOWN:drawing = Trueix, iy = x, yelif event == cv2.EVENT_MOUSEMOVE:if drawing == True:if mode == True:cv2.rectangle(mask, (ix, iy), (x, y), (0, 255, 0), -1)cv2.rectangle(img, (ix, iy), (x, y), (0, 255, 0), -1)else:cv2.circle(mask, (x, y), 5, (0, 255, 0), -1)cv2.circle(img, (x, y), 5, (0, 255, 0), -1)

在这里插入图片描述

在鼠标回调函数中，我们根据不同的鼠标事件进行相应的操作。当按下鼠标左键时，开始画矩形或圆形，记录起始坐标；当鼠标移动时，如果继续按下鼠标左键，根据当前模式画矩形或圆形；当释放鼠标左键时，停止画矩形或圆形，根据当前模式完成最后一个矩形或圆形的绘制。

			elif event == cv2.EVENT_LBUTTONUP:drawing = Falseif mode == True:cv2.rectangle(mask, (ix, iy), (x, y), (0, 255, 0), -1)cv2.rectangle(img, (ix, iy), (x, y), (0, 255, 0), -1)else:cv2.circle(mask, (x, y), 5, (0, 255, 0), -1)cv2.circle(img, (x, y), 5, (0, 255, 0), -1)

4 调用opencv函数

接下来，我们定义了一个waterprint()函数，用于去除图片中的水印。在该函数中，我们首先定义了要查找的颜色范围，然后将图片转换为HSV颜色空间，并使用cv2.inRange()函数根据颜色范围得到一个二值图像。接着，我们使用cv2.dilate()函数对二值图像进行膨胀操作，得到一张掩膜图像。最后，我们使用cv2.inpaint()函数根据掩膜图像进行修复操作，得到最终的去水印结果。

def waterprint(img):global mask#开始操作# 设定要查找的颜色范围lower_green = np.array([50, 100, 100])upper_green = np.array([70, 255, 255])hsv = cv2.cvtColor(mask, cv2.COLOR_BGR2HSV)thresh = cv2.inRange( hsv,lower_green,upper_green) scan = np.ones((5,5),np.uint8)cor = cv2.dilate(thresh,scan,iterations=1)dst = cv2.inpaint(img, cor, 3, cv2.INPAINT_TELEA)

5 绘制蒙版主循环

在主程序中，我们创建了一个窗口，并将回调函数与窗口绑定。然后，我们进入一个循环，不断显示原始图片，并根据按键的操作切换绘制模式、生成掩膜图像，以及退出程序。

while(1):cv2.imshow('image', img)k = cv2.waitKey(1) & 0xFFif k == ord('m'):mode = not modeelif k == ord('s'):mask = np.zeros(img.shape[:2], dtype=np.uint8)contours, hierarchy = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)for contour in contours:x, y, w, h = cv2.boundingRect(contour)cv2.rectangle(mask, (x, y), (x+w, y+h), (0, 0, 0), -1)cv2.imshow('mask', mask)elif k == 27:breakcv2.destroyAllWindows()

6 去水印主循环

在处理视频时，我们首先创建了一个视频编写器，并使用cv2.VideoWriter()函数指定输出视频的格式、帧率和大小。然后，我们使用cv2.VideoCapture()函数读取视频的每一帧，并对每一帧进行去水印操作。最后，我们将处理后的帧写入输出视频，并显示处理后的帧。如果按下键盘上的"s"键，则停止处理并保存输出视频。最后，我们释放资源，关闭窗口。

# 创建视频编写器
fourcc = cv2.VideoWriter_fourcc(*'mp4v')
out = cv2.VideoWriter('output'+datetime.now().strftime("%H-%M-%S")+'.mp4', fourcc, 20.0, (width, height))if pmode=="video":if cap.isOpened():cap.release()cap=cv2.VideoCapture(vieodpath)while(cap.isOpened()):        ret, frame = cap.read()if ret:# 写入输出视频nowaterprint_frame=waterprint(frame)out.write(nowaterprint_frame)# 显示帧cv2.imshow('frame', nowaterprint_frame)if cv2.waitKey(27) & 0xFF == ord('s'):            # 释放资源breakcap.release()out.release()cv2.destroyAllWindows()    
else:nowaterprint_frame=waterprint(img)cv2.imshow('frame', nowaterprint_frame)cv2.waitKey(0)cv2.destroyAllWindows()