实现对图片或者视频增加隐藏水印和提取水印

好久好久没有写博客了,最近看见一个很有意思的文章:小心你的电脑被窃听,就是说在一些公司,截图都会存在水印,方便溯源,然后出于技术的好奇,我在github上搜了一下,还真有相关的github项目:对图片进行水印增加和提取,下面是地址:blind_watermark

1.实现对图片的水印提取

pip install blind-watermark

嵌入水印: blind_watermark --embed --pwd 1234 examples/pic/ori_img.jpeg "watermark text" examples/output/embedded.png

提取水印: blind_watermark --extract --pwd 1234 --wm_shape 111 examples/output/embedded.png

很简单,就是按着命令后走就行,但是问题是:公司的图片里面隐藏的水印的加密方式,我们也不知道啊,我提取了一下我的截图,出现的就是乱码

2.实现对视频的水印添加和提取

这个没什么,就是在上面的基础上,对视频进行分帧,然后对每帧进行水印的添加,然后再加添加了水印的图片合成video,还有对每帧进行水印的提取。

复制代码
import os
from blind_watermark import WaterMark
import cv2
def frames_to_video(frame_folder, output_video_path, frame_rate):
    frame_paths = sorted([os.path.join(frame_folder, f) for f in os.listdir(frame_folder)])
    first_frame = cv2.imread(frame_paths[0])
    height, width, _ = first_frame.shape

    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
    video = cv2.VideoWriter(output_video_path, fourcc, frame_rate, (width, height))

    for frame_path in frame_paths:
        frame = cv2.imread(frame_path)
        video.write(frame)

    video.release()

def embed_watermark_on_frames(input_folder, output_folder, password_img, password_wm, watermark):
    bwm = WaterMark(password_img=password_img, password_wm=password_wm)
    bwm.read_wm(watermark, mode='str')

    for frame_path in os.listdir(input_folder):
        frame_full_path = os.path.join(input_folder, frame_path)
        bwm.read_img(frame_full_path)
        output_frame_path = os.path.join(output_folder, frame_path)
        bwm.embed(output_frame_path)

def video_to_frames(video_path, output_folder):
    cap = cv2.VideoCapture(video_path)
    frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
    frame_rate = int(cap.get(cv2.CAP_PROP_FPS))

    frame_idx = 0
    while cap.isOpened():
        ret, frame = cap.read()
        if not ret:
            break
        frame_path = f"{output_folder}/frame_{frame_idx:04d}.png"
        cv2.imwrite(frame_path, frame)
        frame_idx += 1

    cap.release()
    return frame_count, frame_rate
def extract_watermark_from_frames(frame_folder, password_img, password_wm, wm_shape):
    bwm = WaterMark(password_img=password_img, password_wm=password_wm)
    for frame_path in os.listdir(frame_folder):
        frame_full_path = os.path.join(frame_folder, frame_path)
        watermark = bwm.extract(frame_full_path, wm_shape=wm_shape, mode='str')
        print(f"Watermark from {frame_path}: {watermark}")

# 分解视频
video_path = "test.mp4"
output_frames_folder = "frames"
os.makedirs(output_frames_folder, exist_ok=True)
frame_count, frame_rate = video_to_frames(video_path, output_frames_folder)

# 嵌入水印
watermark = "secret"
password_img = 1
password_wm = 1
output_watermarked_frames = "watermarked_frames"
os.makedirs(output_watermarked_frames, exist_ok=True)
embed_watermark_on_frames(output_frames_folder, output_watermarked_frames, password_img, password_wm, watermark)

# 合成为视频
output_video_path = "watermarked_video.mp4"
frames_to_video(output_watermarked_frames, output_video_path, frame_rate)

# 提取水印
extract_watermark_from_frames(output_watermarked_frames, password_img, password_wm, wm_shape=len(watermark))
相关推荐
源码_V_saaskw13 小时前
JAVA图文短视频交友+自营商城系统源码支持小程序+Android+IOS+H5
java·微信小程序·小程序·uni-app·音视频·交友
肥or胖1 天前
【音视频协议篇】WebRTC 快速入门
ffmpeg·音视频·webrtc
aqi001 天前
FFmpeg开发笔记(七十八)采用Kotlin+Compose的NextPlayer播放器
android·ffmpeg·音视频·直播·流媒体
tang_jian_dong1 天前
springboot + vue3 拉取海康视频点位及播放
spring boot·后端·音视频
青牛科技-Allen1 天前
GC8871刷式直流电机驱动器深度解析:3.6A驱动与内置电流检测技术
单片机·嵌入式硬件·音视频·音响·电动工具·散热风扇·电脑散热风扇
Tracy9732 天前
HF83311_VB1/HF83311Q_VB1:高性能USB HiFi音频解码器固件技术解析
音视频·xmos 模组·xmos 模组固件
_可乐无糖2 天前
AWS WebRTC:我们的业务模式
云计算·音视频·webrtc·aws
Kandiy180253981872 天前
DP4871音频放大芯片3W功率单通道AB类立体声/音频放大器
音视频·音频·视频编解码
is08152 天前
使用 FFmpeg 实现 RTP 音频传输与播放
ffmpeg·音视频