VideoSeal: 동영상 저작권을 보호하는 고급 오픈 소스 동영상 히든 워터마크 삽입 및 추출 도구

1.9K 00

일반 소개

VideoSeal是由Facebook Research开发的开源视频水印工具，旨在提供高效的视频水印嵌入与提取功能。该工具支持最新的开源模型，包含预训练模型、训练代码、推理代码和评估工具，全部在MIT许可下发布。VideoSeal不仅支持视频水印，还适用于图像水印，提供了多种基准模型（如MBRS、CIN、TrustMark和WAM）以供使用、修改和分发。该工具的设计目标是为用户提供一个灵活且高效的视频保护解决方案。

体验地址：https://aidemos.meta.com/videoseal/demo

기능 목록

视频水印嵌入：将水印嵌入到视频中，保护视频版权。
视频水印提取：从视频中提取嵌入的水印信息。
图像水印嵌入：支持将水印嵌入到图像中。
图像水印提取：从图像中提取嵌入的水印信息。
사전 교육 모델：提供多种预训练模型，用户可以直接使用。
训练代码：提供完整的训练代码，用户可以根据需要训练自己的模型。
推理代码：提供推理代码，方便用户进行水印嵌入与提取操作。
评估工具：提供评估工具，帮助用户评估水印嵌入与提取的效果。

도움말 사용

설치 프로세스

Python 설치：确保系统中安装了Python 3.10版本。
安装PyTorch：使用以下命令安装PyTorch及其依赖项：

   pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

安装VideoSeal：克隆VideoSeal仓库并安装依赖项：

   git clone https://github.com/facebookresearch/videoseal.git
cd videoseal
pip install -e .

安装Decord（可选）：用于视频处理的库，推荐安装：

   pip install decord

사용 프로세스

视频水印嵌入

비디오 로드：使用torchvision加载并归一化视频：

   import torchvision
import videoseal
from videoseal.evals.metrics import bit_accuracy
video_path = "assets/videos/1.mp4"
video = torchvision.io.read_video(video_path, output_format="TCHW")
video = video.float() / 255.0

모델 로드：加载VideoSeal模型：

   model = videoseal.load("videoseal")

嵌入水印：将水印嵌入视频：

   outputs = model.embed(video, is_video=True)
video_w = outputs["imgs_w"]
msgs = outputs["msgs"]

视频水印提取

提取水印信息：从水印视频中提取嵌入的信息：

   msg_extracted = model.extract_message(video_w, aggregation="avg", is_video=True)

图像水印嵌入与提取

加载图像：加载并归一化图像：

   img = video[0:1]
outputs = model.embed(img, is_video=False)
img_w = outputs["imgs_w"]
msg_extracted = model.extract_message(img_w, aggregation="avg", is_video=False)