rembg AI Background Removal CLI and Python Library
rembg is a Python tool for automatic image background removal powered by AI models like U2-Net and SAM. It works as a CLI, Python library, HTTP server, or Docker container, supporting CPU and GPU acceleration for batch processing of images and video frames.
Installation
Use the upstream install or setup path that matches your environment:
- pip install "rembg[cpu]" # for library
- pip install "rembg[cpu,cli]" # for library + cli
- pip install "rembg[gpu]" # for library
- pip install "rembg[gpu,cli]" # for library + cli
Requirements and caveats from upstream:
- python: >=3.11, <3.14
- Note: NVIDIA GPUs may require onnxruntime-gpu, CUDA, and cudnn-devel. See #668 for details. If rembg[gpu] doesn't work and you can't insta...
Basic usage or getting-started notes:
text
Choose one of the following backends based on your hardware:
CPU support
Extracted from upstream docs: https://raw.githubusercontent.com/danielgatis/rembg/HEAD/README.md