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.12
Installation
CPU support:
pip install rembg # for library
pip install rembg[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 in your terminal window.
The rembg command has 4 subcommands, one for each input type:
-
ifor files -
pfor folders -
sfor http server -
bfor RGB24 pixel binary stream
You can get help about the main command…
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