comfyui-save-image-watermark

💡 Try it yourself!

cd examples
python3 -m venv venv && source venv/bin/activate && pip install Pillow
python lsb_demo.py aicuty_000011.png
# → Extracts "invisible Hello World!"

Features

Image Logo Watermark

• MASK Support: Accurate alpha blending using LoadImage's MASK output
• Transparent PNG Support: No black/white bleeding in compositing
• Position, Scale, Opacity: Fine-grained control

Text Watermark

• Custom Text: Configurable font size, color, opacity, position
• Dynamic Text: Embed seed values etc. from external nodes
• Tile Mode: Repeat pattern across entire image

Invisible Watermark (Steganography)

• LSB Embedding: Hide messages in the least significant bits
• Extraction Node: Extract embedded hidden messages

Metadata & Provenance

• Workflow Embedding: ComfyUI Core compatible workflow saving
• Content Hash: Generate SHA-256 hash for blockchain provenance
• AICU Metadata: Watermark info, timestamps, etc.

Save Options

• output_folder: Save to ComfyUI output folder
• browser_download: Download via browser
• both: Both (default)

Installation

Via ComfyUI Manager (Recommended)

1. Launch ComfyUI
2. Click Manager button
3. Select Install Custom Nodes
4. Search for watermark or aicu
5. Find comfyui-save-image-watermark and click Install
6. Restart ComfyUI

Via Comfy Registry

comfy node registry-install comfyui-save-image-watermark

Manual Installation

cd ComfyUI/custom_nodes
git clone https://github.com/aicuai/comfyui-save-image-watermark.git

Dependencies

This node uses only ComfyUI's standard libraries, no additional dependencies required.

• Pillow (bundled with ComfyUI)
• NumPy (bundled with ComfyUI)
• PyTorch (bundled with ComfyUI)

Node List

Save Image (Watermark) 💧

Category: AICU/Save

Main node for saving images with watermarks.

Extract Hidden Watermark 🔍

Category: AICU/Watermark

Node for extracting invisible watermarks (steganography).

How to Use in ComfyUI

Adding the Node

1. Right-click on canvas → Add Node
2. Select AICU → Save → Save Image (Watermark) 💧

Node Wiring Diagram

flowchart LR
    subgraph input["Input"]
        KS[KSampler] --> VAE[VAEDecode]
        LI[LoadImage<br/>Logo Image]
    end

    subgraph main["Save Image (Watermark) 💧"]
        direction TB
        I1[/"images (IMAGE)"/]
        I2[/"watermark_image (IMAGE)"/]
        I3[/"watermark_image_mask (MASK)"/]
        I4[/"dynamic_text (STRING)"/]

        O1[\"image (IMAGE)"\]
        O2[\"filename (STRING)"\]
        O3[\"content_hash (STRING)"\]
    end

    subgraph output["Output"]
        PREVIEW[PreviewImage]
        NEXT[Next Node...]
    end

    VAE -->|IMAGE| I1
    LI -->|IMAGE| I2
    LI -->|MASK| I3

    O1 --> PR

Verified Sep 11