Only 7 lines python code to remove background with rembg

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…

View on GitHub

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!!!

Comments

[Samuel Urah Yahaya]

Nice one... Hope to try it out