rembg is a python library which utilizes Neural Network AI models and ONNX to remove the backgrounds from supplied image files. Can be used as a library or a CLI tool.

#image-processing #background-removal #python #library #ai #neuralnet #machine-learning #ml

285 Incheckningar

1 Branches

70 Släpp

Daniel Gatis 1ca14ce058 refactoring		2 år sedan
.github	10a37ba2e2 Install setuptools > 65.5.0	2 år sedan
examples	215cb3e934 add alpha matting	5 år sedan
rembg	1ca14ce058 refactoring	2 år sedan
tests	1ca14ce058 refactoring	2 år sedan
.dockerignore	2bca04e2dd fix docker img	3 år sedan
.editorconfig	7fb6683169 initial	5 år sedan
.gitattributes	fe0823e513 fix build	3 år sedan
.gitignore	e7a8a209db add tests	3 år sedan
Dockerfile	d6a5602e1c fix docker	2 år sedan
LICENSE.txt	7fb6683169 initial	5 år sedan
MANIFEST.in	e7a8a209db add tests	3 år sedan
README.md	1ca14ce058 refactoring	2 år sedan
build-exe	738dd1f59f add build-exe	2 år sedan
onnxruntime-installation-matrix.png	1586c1d5d6 update readme	2 år sedan
pyproject.toml	e7884e2aaf Require younger setuptools	2 år sedan
pytest.ini	e7a8a209db add tests	3 år sedan
rembg.py	2473aa72b9 fix version	3 år sedan
requirements-gpu.txt	1ca14ce058 refactoring	2 år sedan
requirements.txt	1ca14ce058 refactoring	2 år sedan
setup.cfg	fe0823e513 fix build	3 år sedan
setup.py	1ca14ce058 refactoring	2 år sedan
versioneer.py	eba6b2dbf3 fix project layout	3 år sedan

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.11

Installation

CPU support:

pip install rembg

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]

Usage as a cli

After the installation step you can use rembg just typing rembg in your terminal window.

The rembg command has 3 subcommands, one for each input type:

i for files
p for folders
s for http server

You can get help about the main command using:

rembg --help

As well, about all the subcommands using:

rembg <COMMAND> --help

rembg `i`

Used when input and output are files.

Remove the background from a remote image

curl -s http://input.png | rembg i > output.png

Remove the background from a local file

rembg i path/to/input.png path/to/output.png

Remove the background specifying a model

rembg i -m u2netp path/to/input.png path/to/output.png

Remove the background returning only the mask

rembg i -om path/to/input.png path/to/output.png

Remove the background applying an alpha matting

rembg i -a path/to/input.png path/to/output.png

Passing extras parameters

rembg i -m sam -x '{"input_labels": [1], "input_points": [[100,100]]}' path/to/input.png path/to/output.png

rembg `p`

Used when input and output are folders.

Remove the background from all images in a folder

rembg p path/to/input path/to/output

Same as before, but watching for new/changed files to process

rembg p -w path/to/input path/to/output

rembg `s`

Used to start http server.

To see the complete endpoints documentation, go to: http://localhost:5000/docs.

Remove the background from an image url

curl -s "http://localhost:5000/?url=http://input.png" -o output.png

Remove the background from an uploaded image

curl -s -F file=@/path/to/input.jpg "http://localhost:5000"  -o output.png

Usage as a library

Input and output as bytes

from rembg import remove

input_path = 'input.png'
output_path = 'output.png'

with open(input_path, 'rb') as i:
    with open(output_path, 'wb') as o:
        input = i.read()
        output = remove(input)
        o.write(output)

Input and output as a PIL image

from rembg import remove
from PIL import Image

input_path = 'input.png'
output_path = 'output.png'

input = Image.open(input_path)
output = remove(input)
output.save(output_path)

Input and output as a numpy array

from rembg import remove
import cv2

input_path = 'input.png'
output_path = 'output.png'

input = cv2.imread(input_path)
output = remove(input)
cv2.imwrite(output_path, output)

How to iterate over files in a performatic way

from pathlib import Path
from rembg import remove, new_session

session = new_session()

for file in Path('path/to/folder').glob('*.png'):
    input_path = str(file)
    output_path = str(file.parent / (file.stem + ".out.png"))

    with open(input_path, 'rb') as i:
        with open(output_path, 'wb') as o:
            input = i.read()
            output = remove(input, session=session)
            o.write(output)

Usage as a docker

Just replace the rembg command for docker run danielgatis/rembg.

Try this:

docker run danielgatis/rembg i path/to/input.png path/to/output.png

Models

All models are downloaded and saved in the user home folder in the .u2net directory.

The available models are:

u2net (download, source): A pre-trained model for general use cases.
u2netp (download, source): A lightweight version of u2net model.
u2net_human_seg (download, source): A pre-trained model for human segmentation.
u2net_cloth_seg (download, source): A pre-trained model for Cloths Parsing from human portrait. Here clothes are parsed into 3 category: Upper body, Lower body and Full body.
silueta (download, source): Same as u2net but the size is reduced to 43Mb.
isnet-general-use (download, source): A new pre-trained model for general use cases.
sam (encoder, decoder, source): The Segment Anything Model (SAM).