项目地址https://github.com/danielgatis/rembg

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:

i for files
p for folders
s for http server
b for RGB24 pixel binary stream
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 i -m u2net_custom -x '{"model_path": "~/.u2net/u2net.onnx"}' 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/api.

Remove the background from an image url

curl -s "http://localhost:5000/api/remove?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/api/remove"  -o output.png
rembg b
Process a sequence of RGB24 images from stdin. This is intended to be used with another program, such as FFMPEG, that outputs RGB24 pixel data to stdout, which is piped into the stdin of this program, although nothing prevents you from manually typing in images at stdin.

rembg b image_width image_height -o output_specifier
Arguments:

image_width : width of input image(s)
image_height : height of input image(s)
output_specifier: printf-style specifier for output filenames, for example if output-%03u.png, then output files will be named output-000.png, output-001.png, output-002.png, etc. Output files will be saved in PNG format regardless of the extension specified. You can omit it to write results to stdout.
Example usage with FFMPEG:

ffmpeg -i input.mp4 -ss 10 -an -f rawvideo -pix_fmt rgb24 pipe:1 | rembg b 1280 720 -o folder/output-%03u.png
The width and height values must match the dimension of output images from FFMPEG. Note for FFMPEG, the "-an -f rawvideo -pix_fmt rgb24 pipe:1" part is required for the whole thing to work.

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)
To see a full list of examples on how to use rembg, go to the examples page.