WatermarkRemover-AI: An Open-Source AI Watermark Removal Tool 191

小编 ·

WatermarkRemover-AI: An Open-Source AI Watermark Removal Tool

WatermarkRemover-AI is an open-source, practical AI watermark removal tool. It uses Microsoft's Florence-2 technology to detect watermarks and the LaMA model to inpaint them, effectively removing all kinds of watermarks. It supports both single-image and batch processing, delivers natural-looking results, and offers multiple output formats. It runs smoothly even without a GPU, and provides a one-click installation script and an easy-to-use interface.

WatermarkRemover-AI: An Open-Source AI Watermark Removal Tool

What is WatermarkRemover-ai?

WatermarkRemover-AI is an open-source, practical AI watermark-removal tool. It uses Microsoft's Florence-2 technology to detect watermarks and the LaMA model to perform inpainting, effectively removing all kinds of watermarks. It supports both single-image processing and batch operations, delivers natural-looking repair results, and supports multiple output formats. It runs smoothly without requiring a GPU, and also provides a one-click installation script along with a simple and easy-to-use interface.

WatermarkRemover-AI: A powerful AI watermark removal tool.webp

WatermarkRemover-AI Main Features

Dual-mode operation:

  • Supports processing a single image or images in an entire directory.

Advanced watermark detection:

  • Uses Florence-2's open-vocabulary detection capability to accurately identify watermarks.

Seamless inpainting:

  • Uses the LaMA model for context-aware inpainting, naturally filling in the areas where watermarks have been removed.

Customizable output:

  • The maximum bounding box size for watermark detection can be configured.

  • The transparency of the watermark area can be set.

  • Output can be forced into a specific format (PNG, WEBP, JPG).

Progress tracking:

  • Provides real-time progress updates in both GUI and CLI modes.

Dark mode support:

  • The GUI automatically adapts to dark mode based on system settings.

Efficient resource management:

  • Supports GPU acceleration using CUDA (optional).

WatermarkRemover-AI: A powerful AI watermark removal tool.webp

WatermarkRemover-AI Technical Architecture

  • Florence-2: Used for watermark detection, identifying watermarks through open-vocabulary object detection.

  • LaMA: Used for inpainting, context-aware filling of areas where watermarks have been removed.

  • PyQt6: Used to build a user-friendly graphical interface.

How to Install WatermarkRemover-AI

Clone the repository:

git clone https://github.com/D-Ogi/WatermarkRemover-AI.git
cd WatermarkRemover-AI

Run the installation script:

  • Use the setup.sh script to automatically set up the environment, install dependencies, and launch the GUI application.

  • The CLI mode can also be used directly via the command line.

Manually activating the environment:

If you need to manually activate the environment, you can run:

conda activate py312aiwatermark

How to Use WatermarkRemover-AI

Graphical Interface (GUI):

  • After launching the GUI, you can choose to process a single image or an entire directory.

  • You can set the input/output paths and configure options (such as transparency, bounding box size, etc.).

  • Click the "Start" button to begin processing.

Command Line Interface (CLI):

  • Basic command format:

python remwm.py input_path output_path [options]
  • Common options:

  • --overwrite: Overwrite existing files.

  • --transparent: Set the watermark area to transparent.

  • --max-bbox-percent: Set the maximum bounding box size for watermark detection (default is 10%).

  • --force-format: Force the output to a specific format (PNG, WEBP, or JPG).

Example command

python remwm.py ./input_images ./output_images --overwrite --transparent --max-bbox-percent=15 --force-format=PNG

github: https://github.com/D-Ogi/WatermarkRemover-AI