Resolve issue, update README.md

MillionMillionLi · MillionMillionLi · commit 2a6f06719fc9 · 2026-04-01T20:09:14.000+08:00
diff --git a/.github/workflows/docker-publish.yml b/.github/workflows/docker-publish.yml
@@ -50,7 +50,7 @@ jobs:
         id: meta
         uses: docker/metadata-action@v5
         with:
-          images: millionmillionli/aigc-identification-toolkit
+          images: millionmillionli/aigc-watermark-toolkit
           tags: |
             # 始终使用 latest 标签
             type=raw,value=latest
@@ -86,9 +86,9 @@ jobs:
           echo "" >> $GITHUB_STEP_SUMMARY
           echo "**拉取镜像:**" >> $GITHUB_STEP_SUMMARY
           echo "\`\`\`bash" >> $GITHUB_STEP_SUMMARY
-          echo "docker pull millionmillionli/aigc-identification-toolkit:latest" >> $GITHUB_STEP_SUMMARY
+          echo "docker pull millionmillionli/aigc-watermark-toolkit:latest" >> $GITHUB_STEP_SUMMARY
           echo "\`\`\`" >> $GITHUB_STEP_SUMMARY
           echo "" >> $GITHUB_STEP_SUMMARY
-          echo "**DockerHub 链接:** [millionmillionli/aigc-identification-toolkit](https://hub.docker.com/r/millionmillionli/aigc-identification-toolkit)" >> $GITHUB_STEP_SUMMARY
+          echo "**DockerHub 链接:** [millionmillionli/aigc-watermark-toolkit](https://hub.docker.com/r/millionmillionli/aigc-watermark-toolkit)" >> $GITHUB_STEP_SUMMARY
           echo "" >> $GITHUB_STEP_SUMMARY
           echo "**提交 SHA:** \`${{ github.sha }}\`" >> $GITHUB_STEP_SUMMARY
diff --git a/.gitignore b/.gitignore
@@ -99,4 +99,5 @@ CLAUDE.md
 GEMINI.md
 temp/
 class/
-*.pdf
+# *.pdf
+materials/
diff --git a/Dockerfile b/Dockerfile
@@ -1,5 +1,5 @@
 # ====================================
-# AIGC-Identification-Toolkit Docker配置
+# AIGC-Watermark-Toolkit Docker配置
 # 多模态水印Benchmark系统
 # ====================================
 
diff --git a/README.md b/README.md
@@ -52,6 +52,7 @@
 
 ## 📑 目录
 
+- [效果展示](#效果展示)
 - [开始使用](#开始使用)
   - [安装](#安装)
 - [使用方法](#使用方法)
@@ -62,6 +63,40 @@
 
 ---
 
+## 🎬 效果展示
+
+### 图像标识效果
+
+<table>
+  <tr>
+    <th align="center">原始图像</th>
+    <th align="center">隐水印图像</th>
+    <th align="center">显式标识图像</th>
+  </tr>
+  <tr>
+    <td align="center"><img src="docs/assets/demo/original_image.png" width="260"/></td>
+    <td align="center"><img src="docs/assets/demo/watermarked_image.png" width="260"/></td>
+    <td align="center"><img src="docs/assets/demo/visible_marked_image.png" width="260"/></td>
+  </tr>
+</table>
+
+### 视频标识效果
+
+<table>
+  <tr>
+    <th align="center">原始视频</th>
+    <th align="center">隐水印视频</th>
+    <th align="center">显式标识视频</th>
+  </tr>
+  <tr>
+    <td align="center"><video src="docs/assets/demo/original_video.mp4" width="260" controls></video></td>
+    <td align="center"><video src="docs/assets/demo/watermarked_video.mp4" width="260" controls></video></td>
+    <td align="center"><video src="docs/assets/demo/visible_marked_video.mp4" width="260" controls></video></td>
+  </tr>
+</table>
+
+---
+
 ## 开始使用
 
 ### 安装
@@ -71,8 +106,8 @@
 1. 克隆仓库
 
    ```bash
-   git clone --recurse-submodules https://github.com/MillionMillionLi/AIGC-Identification-Toolkit.git
-   cd AIGC-Identification-Toolkit
+   git clone --recurse-submodules https://github.com/MillionMillionLi/AIGC-Watermark-Toolkit.git
+   cd AIGC-Watermark-Toolkit
    ```
 
 2. 安装核心依赖
@@ -119,7 +154,7 @@
 
 1. 拉取镜像
    ```bash
-   docker pull millionmillionli/aigc-identification-toolkit:latest
+   docker pull millionmillionli/aigc-watermark-toolkit:latest
    ```
 
 2. 运行容器
@@ -132,7 +167,7 @@
      -e PYTHONPATH=/app \
      -e CUDA_VISIBLE_DEVICES=0 \
      --restart unless-stopped \
-     millionmillionli/aigc-identification-toolkit:latest \
+     millionmillionli/aigc-watermark-toolkit:latest \
      tail -f /dev/null
    ```
 
diff --git a/README_EN.md b/README_EN.md
@@ -52,6 +52,7 @@ This project provides an all-in-one open-source identification toolkit. Supporti
 
 ## 📑 Table of Contents
 
+- [Demo](#demo)
 - [Getting Started](#getting-started)
   - [Installation](#installation)
 - [Usage](#usage)
@@ -62,6 +63,40 @@ This project provides an all-in-one open-source identification toolkit. Supporti
 
 ---
 
+## 🎬 Demo
+
+### Image Watermarking Effect
+
+<table>
+  <tr>
+    <th align="center">Original Image</th>
+    <th align="center">Hidden Watermark Image</th>
+    <th align="center">Visible Mark Image</th>
+  </tr>
+  <tr>
+    <td align="center"><img src="docs/assets/demo/original_image.png" width="260"/></td>
+    <td align="center"><img src="docs/assets/demo/watermarked_image.png" width="260"/></td>
+    <td align="center"><img src="docs/assets/demo/visible_marked_image.png" width="260"/></td>
+  </tr>
+</table>
+
+### Video Watermarking Effect
+
+<table>
+  <tr>
+    <th align="center">Original Video</th>
+    <th align="center">Hidden Watermark Video</th>
+    <th align="center">Visible Mark Video</th>
+  </tr>
+  <tr>
+    <td align="center"><video src="docs/assets/demo/original_video.mp4" width="260" controls></video></td>
+    <td align="center"><video src="docs/assets/demo/watermarked_video.mp4" width="260" controls></video></td>
+    <td align="center"><video src="docs/assets/demo/visible_marked_video.mp4" width="260" controls></video></td>
+  </tr>
+</table>
+
+---
+
 ## Getting Started
 
 ### Installation
@@ -71,8 +106,8 @@ This project provides an all-in-one open-source identification toolkit. Supporti
 1. Clone the repository
 
    ```bash
-   git clone --recurse-submodules https://github.com/MillionMillionLi/AIGC-Identification-Toolkit.git
-   cd AIGC-Identification-Toolkit
+   git clone --recurse-submodules https://github.com/MillionMillionLi/AIGC-Watermark-Toolkit.git
+   cd AIGC-Watermark-Toolkit
    ```
 
 2. Install core dependencies
@@ -119,7 +154,7 @@ This project provides an all-in-one open-source identification toolkit. Supporti
 
 1. Pull the image
    ```bash
-   docker pull millionmillionli/aigc-identification-toolkit:latest
+   docker pull millionmillionli/aigc-watermark-toolkit:latest
    ```
 
 2. Run the container
@@ -132,7 +167,7 @@ This project provides an all-in-one open-source identification toolkit. Supporti
      -e PYTHONPATH=/app \
      -e CUDA_VISIBLE_DEVICES=0 \
      --restart unless-stopped \
-     millionmillionli/aigc-identification-toolkit:latest \
+     millionmillionli/aigc-watermark-toolkit:latest \
      tail -f /dev/null
    ```
 
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -7,7 +7,7 @@ services:
   # ====================================
   toolkit:
     container_name: aigc-watermark-toolkit
-    image: millionmillionli/aigc-identification-toolkit:latest
+    image: millionmillionli/aigc-watermark-toolkit:latest
 
     # ====================================
     # GPU支持（NVIDIA Docker）
diff --git a/docs/guides/ENVIRONMENT_SETUP.md b/docs/guides/ENVIRONMENT_SETUP.md
@@ -1,6 +1,6 @@
 # 环境配置指南
 
-本指南说明如何配置AIGC Identification Toolkit的运行环境，包括模型缓存路径、离线模式和跨平台支持。
+本指南说明如何配置AIGC Watermark Toolkit的运行环境，包括模型缓存路径、离线模式和跨平台支持。
 
 ## 目录
 
diff --git a/docs/guides/GUIDE_Docker.md b/docs/guides/GUIDE_Docker.md
@@ -219,7 +219,7 @@ sudo docker run --rm --gpus all nvidia/cuda:11.8.0-base-ubuntu22.04 nvidia-smi
 │   │   └── ...
 │   └── transformers/            # 👈 Transformers缓存
 │
-└── AIGC-Identification-Toolkit/ # 👈 项目根目录
+└── AIGC-Watermark-Toolkit/ # 👈 项目根目录
     ├── config/                  # 👈 配置文件
     │   └── default_config.yaml  # 统一配置文件
     ├── src/                     # 👈 源代码
@@ -290,8 +290,8 @@ environment:
 
 ```bash
 cd /fs-computility/wangxuhong/limeilin/
-git clone --recurse-submodules https://github.com/MillionMillionLi/AIGC-Identification-Toolkit.git
-cd AIGC-Identification-Toolkit
+git clone --recurse-submodules https://github.com/MillionMillionLi/AIGC-Watermark-Toolkit.git
+cd AIGC-Watermark-Toolkit
 ```
 
 > 💡 `--recurse-submodules` 会自动克隆VINE和AudioMarkBench子模块
diff --git a/docs/guides/assets/demo/docs/assets/demo/watermarked_image.jpg.png b/docs/guides/assets/demo/docs/assets/demo/watermarked_image.jpg.png
diff --git a/docs/guides/assets/demo/original_image.png b/docs/guides/assets/demo/original_image.png
diff --git a/docs/guides/assets/demo/original_video.mp4 b/docs/guides/assets/demo/original_video.mp4
diff --git a/docs/guides/assets/demo/visible_marked_image.png b/docs/guides/assets/demo/visible_marked_image.png
diff --git a/docs/guides/assets/demo/visible_marked_video.mp4 b/docs/guides/assets/demo/visible_marked_video.mp4
diff --git a/docs/guides/assets/demo/watermarked_image.png b/docs/guides/assets/demo/watermarked_image.png
diff --git a/docs/guides/assets/demo/watermarked_video.mp4 b/docs/guides/assets/demo/watermarked_video.mp4
diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,5 @@
 # ====================================
-# AIGC-Identification-Toolkit Requirements
+# AIGC-Watermark-Toolkit Requirements
 # ====================================
 
 # ============== 核心深度学习框架 ==============
diff --git a/setup.py b/setup.py
@@ -7,7 +7,7 @@
     requirements = [line.strip() for line in fh if line.strip() and not line.startswith("#")]
 
 setup(
-    name="AIGC-Identification-Toolkit",
+    name="AIGC-Watermark-Toolkit",
     version="0.1.0",
     author="lml",
     author_email="",
diff --git a/src/unified/unified_engine.py b/src/unified/unified_engine.py
@@ -189,7 +189,7 @@ def _get_video_watermark(self) -> VideoWatermark:
         """获取视频水印处理器"""
         if self._video_watermark is None:
             from ..video_watermark.video_watermark import create_video_watermark
-            self._video_watermark = create_video_watermark()
+            self._video_watermark = create_video_watermark(config_path=self.config_path)
         return self._video_watermark
     
     def embed(self, content: str, message: str, modality: str, operation: str = 'watermark', **kwargs) -> Any:
diff --git a/src/utils/path_manager.py b/src/utils/path_manager.py
@@ -1,5 +1,5 @@
 """
-Unified Path Manager for AIGC Identification Toolkit
+Unified Path Manager for AIGC Watermark Toolkit
 
 This module provides cross-platform path resolution for model caches and data directories.
 It follows the priority order:
diff --git a/src/video_watermark/video_watermark.py b/src/video_watermark/video_watermark.py
@@ -125,7 +125,8 @@ def _ensure_video_generator(self) -> WanVideoGenerator:
     def _ensure_watermark_wrapper(self) -> VideoSealWrapper:
         """确保水印包装器已初始化"""
         if self.watermark_wrapper is None:
-            self.watermark_wrapper = VideoSealWrapper(self.device)
+            model_card = self.config.get('videoseal', {}).get('model_card', 'videoseal_1.0')
+            self.watermark_wrapper = VideoSealWrapper(self.device, model_card=model_card)
         return self.watermark_wrapper
     
     def generate_video_with_watermark(
@@ -523,19 +524,31 @@ def clear_cache(self):
 # 方便的工厂函数
 def create_video_watermark(
     cache_dir: Optional[str] = None,
-    device: Optional[str] = None
+    device: Optional[str] = None,
+    config_path: Optional[str] = None
 ) -> VideoWatermark:
     """
     创建视频水印工具的快捷函数
 
     Args:
         cache_dir: 模型缓存目录（None则使用环境变量或默认路径）
         device: 计算设备
+        config_path: 配置文件路径，如果为None则使用默认配置
 
     Returns:
         VideoWatermark: 视频水印工具实例
     """
-    return VideoWatermark(cache_dir=cache_dir, device=device)
+    config = {}
+    if config_path:
+        import yaml
+        with open(config_path, 'r', encoding='utf-8') as f:
+            full_config = yaml.safe_load(f)
+        config = full_config.get('video_watermark', {})
+        if not cache_dir:
+            cache_dir = config.get('cache_dir')
+        if not device:
+            device = config.get('device')
+    return VideoWatermark(cache_dir=cache_dir, device=device, config=config)
 
 
 if __name__ == "__main__":
diff --git a/src/video_watermark/videoseal_wrapper.py b/src/video_watermark/videoseal_wrapper.py
@@ -29,14 +29,16 @@
 class VideoSealWrapper:
     """VideoSeal水印算法包装器"""
     
-    def __init__(self, device: Optional[str] = None):
+    def __init__(self, device: Optional[str] = None, model_card: str = "videoseal_1.0"):
         """
         初始化VideoSeal包装器
-        
+
         Args:
             device: 计算设备 ('cuda', 'cpu', 或None自动选择)
+            model_card: VideoSeal模型卡名称
         """
         self.device = device or ('cuda' if torch.cuda.is_available() else 'cpu')
+        self.model_card = model_card
         self.model = None
         
         # 设置日志
@@ -63,7 +65,7 @@ def _load_model(self):
             os.chdir(videoseal_dir)
             try:
                 # 使用videoseal的默认模型加载方式（指定模型卡）
-                self.model = videoseal.load("videoseal_1.0")
+                self.model = videoseal.load(self.model_card)
             finally:
                 # 恢复工作目录
                 os.chdir(current_dir)
