【python】去除水印的几种方式

hzxOnlineOk

已于 2024-04-28 14:21:10 修改

阅读量8.6k

点赞数 12

文章标签： python 开发语言

于 2024-04-28 14:15:51 首次发布

本文链接：https://blog.csdn.net/hzxOnlineOk/article/details/138276551

版权

本文介绍了如何使用Python通过FFmpeg的delogo函数和OpenCV库去除视频中的水印，包括使用FFmpeg命令行和OpenCV的图像处理技术，如seamlessClone和matchTemplate函数，提供了详细的代码示例和两种方法的使用场景.

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1、python调用FFMEPG的delogo函数去除水印

要使用Python调用FFmpeg的delogo filter去除视频水印，你需要使用subprocess模块运行FFmpeg命令。以下是一个简单的Python脚本示例：

import subprocess
 
def remove_watermark(input_video, output_video, logo_x, logo_y, logo_width, logo_height):
    # 构建FFmpeg命令
    command = [
        'ffmpeg',
        '-i', input_video,
        '-vf', f'delogo=x={
     logo_x}:y={
     logo_y}:w={
     logo_width}:h={
     logo_height}',
        output_video
    ]
    
    # 运行FFmpeg命令
    subprocess.run(command)
 
# 使用函数去除水印
remove_watermark('input.mp4', 'output.mp4', 10, 10, 100, 100)

使用说明视频：https://www.bilibili.com/video/BV1Jg4y1e7JJ/?spm_id_from=pageDriver

2、使用opencv-python库来处理视频帧

要在Python中去除视频水印，可以使用opencv-python库来处理视频帧，并结合图像处理技术，如图像修复或者图层混合。以下是一个简单的示例，演示如何使用OpenCV去除静态图像水印：

import cv2
import numpy as np
 
def remove_watermark(video_path, watermark_path, output_path):
    # 读取视频和水印图像
    cap = cv2.VideoCapture(video_path)
    watermark = cv2.imread(watermark_path, cv2.IMREAD_UNCHANGED)
    watermark = cv2.cvtColor(watermark, cv2.COLOR_BGR2GRAY)
    watermark = cv2.GaussianBlur(watermark, (5, 5), 0)
 
    # 获取水印的mask
    _, mask = cv2.threshold(watermark, 1, 255, cv2.THRESH_BINARY_INV)
 
    while True:
        ret, frame = cap.read()
        if not ret:
            break
 
        # 将水印区域替换为视频帧的背景
        frame_gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
        seamless_clone = cv2.seamlessClone(watermark, frame, mask, center, cv2.NORMAL_CLONE)
 
        # 写入去水印后的视频帧
        output_video.write(seamless_clone)
 
    cap.release()
    output_video.release()
 
# 使用函数去除视频中的水印
remove_watermark('input_video.mp4', 'watermark.png', 'output_video.mp4')

请注意，这个示例使用了seamlessClone函数，它要求水印区域的中心与背景相匹配，并且假设水印背景是纯色或者与视频背景融合得当。如果这些条件不满足，可能需要更复杂的图像处理技术，例如图像修复或深度学习去水印方法。

示例2
下面是使用OpenCV去除水印的Python代码示例：

import cv2
import numpy as np
 
# 读取视频和水印图像
video_path = 'video_with_watermark.mp4'
watermark_path = 'watermark.png'
cap = cv2.VideoCapture(video_path)
 
# 读取视频的宽、高和帧数
frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
frame_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
fps = cap.get(cv2.CAP_PROP_FPS)
 
# 读取水印图像
watermark = cv2.imread(watermark_path, cv2.IMREAD_UNCHANGED)
watermark_gray = cv2.cvtColor(watermark, cv2.COLOR_BGR2GRAY)
 
# 创建输出视频
out = cv2.VideoWriter('video_without_watermark.mp4', cv2.VideoWriter_fourcc(*'XVID'), fps, (frame_width