Extensions/ComfyUI Watermark Detection Node
ComfyUI Extension

ComfyUI Watermark Detection Node

This custom node for ComfyUI provides watermark detection capabilities using a YOLO model trained by [a/fancyfeast](https://huggingface.co/fancyfeast), the creator of…

By lrzjason·Created about a year ago·Updated about a year ago· 51
lrzjason/ComfyUI-Watermark-Detection
Nodes2
On cloudLocal install
Categorywatermark_detection
Stars51
Updatedabout a year ago

Nodes (2)

Detect Watermarks
watermark_detection
Load Watermark Detector
watermark_detection
Readme

ComfyUI Watermark Detection Node

This custom node for ComfyUI provides watermark detection capabilities using a YOLO model trained by fancyfeast, the creator of JoyCaption. The model is originally hosted at Hugging Face Space.

Features

  • Detects watermarks in images using YOLOv11
  • Returns both the detection visualization and a binary mask
  • Adjustable confidence threshold for detections

Installation

  1. Place this folder in your ComfyUI/custom_nodes/ directory
  2. The model will be automatically downloaded from Hugging Face on first use

Usage

  1. Add a Load Watermark Detector node to load the model
  2. Connect your image to the Detect Watermarks node
  3. Adjust the confidence threshold as needed (default: 0.5)
  4. The node outputs:
    • Detection visualization (IMAGE)
    • Binary mask of detected watermarks (MASK)

Model Details

  • Model: YOLOv11 trained on watermark detection
  • Original Author: fancyfeast (JoyCaption creator)
  • Model Source: Hugging Face
  • Re-hosted at Hugging Face

Notes

  • The model will be downloaded to ComfyUI/models/yolo/ automatically
  • Higher confidence thresholds will result in fewer but more certain detections