Sora 水印移除 closerAIsorawater ComfyUI 节点

本节点是基于 sora视频 的 AI 检测逻辑，专为 ComfyUI 设计，用于自动检测和移除视频中的Sora水印。

节点采用逐帧动态蒙版技术，可处理在视频中循环出现（如左上、左下、中右）的水印。

快速安装

请严格按照以下步骤操作，顺序不可颠倒。

1. 安装 Python 依赖库

本节点需要 cryptography 和 ultralytics (YOLOv8) 两个核心库。

1. 打开您的 ComfyUI 启动器（run_nvidia_gpu.bat 或 run_cpu.bat）。
2. 在启动菜单中，点击“Install Pythons deps”（或类似的管理/安装依赖项的选项）。
3. 在弹出的命令行窗口中，依次输入并执行以下命令：Bashpip install cryptography pip install ultralytics
4. 等待安装完成后，关闭此窗口。

2. 下载并放置 AI 模型

本节点需要两个 AI 模型才能工作：

• 水印检测模型 (YOLO):
  • 您需要一个专门用于检测 Sora 水印的 YOLO 模型（通常是 .pt 文件，例如从 SoraWatermarkCleaner 项目中获取的 best.pt）。
  • 请将此 .pt 文件放入您的 ComfyUI 文件夹：ComfyUI/models/yolo/
  • (如果 yolo 文件夹不存在，请手动创建它)
• 图像修复模型 (LaMa):
  • 您需要一个 LaMa 修复模型（通常是 .pt 或 .pth 文件）。
  • 请将此模型文件放入您的 ComfyUI 文件夹：ComfyUI/models/lama/
  • (如果 lama 文件夹不存在，请手动创建它)

3. 安装节点文件

1. 前往您的 ComfyUI custom_nodes 文件夹： E:\comfyUI\ComfyUI-aki-v1.4\ComfyUI-aki-v1.4\custom_nodes\ (请替换为您的实际路径)
2. （重要） 确保此文件夹中没有任何旧的 closerAI_sorawater_nodes.py 文件。
3. 将您最终的、加密版的 closerAI_sorawater_nodes.py 文件复制到此 custom_nodes 文件夹中。

4. 重启 ComfyUI

彻底关闭并重新启动您的 ComfyUI。

启动时，请查看控制台日志。如果您看到以下信息，则代表加载成功： ✅ CloserAI SoraWater 节点 (已加密) 加载成功。

使用指南

1. 加载节点

1. 在 ComfyUI 工作区中双击。
2. 搜索 “Sora水印移除 (CloserAI 优化版)” 并添加它。

2. 连接输入

• frames (IMAGE): 连接您的视频帧来源，例如 VideoHelperSuite (VHS) 的 "Load Video" 节点。
• fps (FLOAT): 连接视频的fps（帧率）输入，这决定了输出视频的播放速度。

3. 配置参数（核心）

• 检测模式:
  • automatic: 默认选项。节点将使用 AI 自动检测每一帧的水印。（推荐）
  • manual_mask_input: 手动模式。您必须从 optional_mask 接口连接一个您自己绘制的蒙版。
• Lama修复模型:
  • 从下拉菜单中，选择您在安装步骤 2 中放入 models/lama/ 文件夹的修复模型。
• 输出文件名:
  • 设置您希望保存的视频名称，例如 my_video_no_watermark.mp4。视频将保存在 ComfyUI 的 output 文件夹中。
• 保存视频:
  • 勾选此项，节点在处理完毕后会自动将清理后的帧合成为视频文件。

4. 自动模式参数（重要）

当检测模式为 automatic 时，以下参数非常关键：

• YOLO检测模型:
  • 从下拉菜单中，选择您在安装步骤 2 中放入 models/yolo/ 文件夹的水印检测模型。
• YOLO置信度阈值:
  • 这是最重要的参数。它决定了 AI 的“敏感度”。
  • 值太高 (如 0.8)：AI 可能过于“严格”，导致找不到水印（final_mask 输出全黑）。
  • 值太低 (如 0.05)：AI 可能过于“敏感”，把视频中的人物、高光等也当成水印，导致蒙版范围错误（例如蒙版在画面中央）。
  • 推荐：从 0.15 或 0.2 开始测试。如果蒙版错误，请尝试微调此值。
• 蒙版膨胀:
  • 在检测到的水印蒙版基础上，向外扩张几个像素。这有助于消除修复后可能出现的轻微“边缘”。推荐使用 5 到 10。

5. 输出

• cleaned_frames (IMAGE): 已清除水印的视频帧。您可以连接一个 "Save Animated Webp" 或 "Video Combine" 节点来预览。
• final_mask (MASK): 这是一个预览蒙版，它显示了在整个视频中所有被检测到的水印位置的“并集”（所有位置的合集）。它用于帮您调试 YOLO置信度阈值 是否正确。

主题授权提示：请在后台主题设置-主题授权-激活主题的正版授权，授权购买：RiTheme官网