danielgatis/rembg
Rembg is a tool to remove images background.
repo name | danielgatis/rembg |
repo link | https://github.com/danielgatis/rembg |
homepage | |
language | Python |
size (curr.) | 15156 kB |
stars (curr.) | 310 |
created | 2020-08-10 |
license | MIT License |
Rembg
Rembg is a tool to remove images background. That is it.
Requirements
- python >= 3.8
Installation
Install it from pypi
pip install rembg
Usage as a cli
Remove the background from a remote image
curl -s http://input.png | rembg > output.png
Remove the background from a local file
rembg -o path/to/output.png path/to/input.png
Remove the background from all images in a folder
rembg -p path/to/inputs
Usage as a server
Start the server
rembg-server
Open your browser to
http://localhost:5000?url=http://image.png
Also you can send the file as a FormData (multipart/form-data):
<form action="http://localhost:5000" method="post" enctype="multipart/form-data">
<input type="file" name="file"/>
<input type="submit" value="upload"/>
</form>
Usage as a library
In app.py
import sys
from rembg.bg import remove
sys.stdout.buffer.write(remove(sys.stdin.buffer.read()))
Then run
cat input.png | python app.py > out.png
Advance usage
Sometimes it is possible to achieve better results by turning on alpha matting. Example:
curl -s http://input.png | rembg -a -ae 15 > output.png
References
- https://arxiv.org/pdf/2005.09007.pdf
- https://github.com/NathanUA/U-2-Net
- https://github.com/pymatting/pymatting
License
Copyright (c) 2020-present Daniel Gatis
Licensed under MIT License
Buy me a coffee
Liked some of my work? Buy me a coffee (or more likely a beer)