danielgatis
/
rembg
Rembg is a tool to remove images background
Rembg
Rembg is a tool to remove images background.
If this project has helped you, please consider making a donation.
Sponsor
PhotoRoom Remove Background API
https://photoroom.com/api
Fast and accurate background remover API
Requirements
python: >3.7, <3.13
Installation
If you have onnxruntime already installed, just install rembg:
pip install rembg # for library
pip install "rembg[cli]" # for library + cli
Otherwise, install rembg with explicit CPU/GPU support.
CPU support:
pip install rembg[cpu] # for library
pip install "rembg[cpu,cli]" # for library + cli
GPU support:
First of all, you need to check if your system supports the onnxruntime-gpu.
Go to https://onnxruntime.ai and check the installation matrix.
If yes, just run:
pip install "rembg[gpu]" # for library
pip install "rembg[gpu,cli]" # for library + cli
Usage as a cli
After the installation step you can use rembg just typing rembg…
We can run gpu version on Google colab
!pip install rembg[gpu]
If you want to run the script on your local machine without a gpu, you will need to install 2 packages. If you have OpenCV, you can use it instead of Pillow.
pip install rembg
pip install -U Pillow
from rembg import remove
from PIL import Image
input_path = 'input.jpg' # input image path
output_path = 'output.png' # output image path
input = Image.open(input_path) # load image
output = remove(input) # remove background
output.save(output_path) # save image
Incredibly easy!!!
Top comments (1)
Nice one... Hope to try it out