havetofighting / rembg

Rembg is a tool to remove images background.

Rembg

Rembg is a tool to remove images background. That is it.

Requirements

• python 3.8 or newer

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

Usage as a docker

Just run

  curl -s http://input.png | docker run -i -v ~/.u2net:/root/.u2net danielgatis/rembg:latest > output.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

Original	Without alpha matting	With alpha matting (-a -ae 15)

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)