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, img

    if event == cv2.EVENT_LBUTTONDOWN:
        drawing = True
        ix, iy = x, y

    elif 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 = False
			        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)

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) & 0xFF
    if k == ord('m'):
        mode = not mode
    elif 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:
        break

cv2.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'):            
            
            # 释放资源
            
            break
        
    cap.release()
    out.release()
    cv2.destroyAllWindows()    
else:
    nowaterprint_frame=waterprint(img)
    cv2.imshow('frame', nowaterprint_frame)
    cv2.waitKey(0)
    cv2.destroyAllWindows()