Add files via upload

2024-11-14 15:28:59 -08:00
parent d6bdb5762a
commit bd94f829a6
1 changed files with 70 additions and 1 deletions
@@ -1 +1,70 @@
-# ComfyUI-RMBG
+# ComfyUI-RMBG
+
+A ComfyUI node for removing image backgrounds using RMBG-2.0, a state-of-the-art background removal model developed by BRIA AI.
+
+## Features
+
+RMBG-2.0 is built on the innovative BiRefNet (Bilateral Reference Network) architecture, offering:
+- High accuracy in complex environments
+- Precise edge detection and preservation
+- Excellent handling of fine details
+- Support for multiple objects in a single image
+
+## Installation
+
+1. Clone this repository to your ComfyUI custom_nodes folder:
+```bash
+cd ComfyUI/custom_nodes
+git clone https://github.com/1038lab/ComfyUI-RMBG
+```
+
+2. Download RMBG model:
+- Download from [RMBG-2.0](https://huggingface.co/briaai/RMBG-2.0)
+- Place the model files in `ComfyUI/models/RMBG/RMBG-2.0/` directory
+
+## Usage
+
+### Basic Usage
+1. Load `🧽 RMBG (Remove Background)` node from the `🧪AILab/🧽RMBG` category
+2. Connect an image to the input
+3. Get two outputs:
+   - IMAGE: Processed image with transparent background
+   - MASK: Binary mask of the foreground
+
+### Parameters
+- `sensitivity`: Controls the background removal sensitivity (0.0-1.0)
+- `process_res`: Processing resolution (512-2048, step 128)
+- `mask_blur`: Blur amount for the mask (0-64)
+- `mask_offset`: Adjust mask edges (-20 to 20)
+
+## About RMBG-2.0
+
+RMBG-2.0 is developed by BRIA AI and uses the BiRefNet architecture which includes:
+
+- **Localization Module (LM)**: Generates semantic maps for primary image areas
+- **Restoration Module (RM)**: Performs precise boundary restoration using:
+  - Original Reference: Provides general background context
+  - Gradient Reference: Focuses on edges and fine details
+
+The model is trained on a diverse dataset of over 15,000 high-quality images, ensuring:
+- Balanced representation across different image types
+- High accuracy in various scenarios
+- Robust performance with complex backgrounds
+
+## Requirements
+- ComfyUI
+- Python 3.10+
+- Required packages (automatically installed):
+  - torch>=2.0.0
+  - torchvision>=0.15.0
+  - Pillow>=9.0.0
+  - numpy>=1.22.0
+  - transformers>=4.30.0
+  - safetensors>=0.3.0
+
+## Credits
+- RMBG-2.0: https://huggingface.co/briaai/RMBG-2.0
+- Created by: [1038 Lab](https://github.com/1038lab)
+
+## License
+MIT License