rembg


Namerembg JSON
Version 2.0.56 PyPI version JSON
download
home_pagehttps://github.com/danielgatis/rembg
SummaryRemove image background
upload_time2024-03-12 22:54:58
maintainer
docs_urlNone
authorDaniel Gatis
requires_python>=3.8, <3.13
license
keywords remove background u2net
VCS
bugtrack_url
requirements No requirements were recorded.
Travis-CI No Travis.
coveralls test coverage No coveralls.
            # Rembg

[![Downloads](https://img.shields.io/pypi/dm/rembg.svg)](https://img.shields.io/pypi/dm/rembg.svg)
[![License](https://img.shields.io/badge/License-MIT-blue.svg)](https://img.shields.io/badge/License-MIT-blue.svg)
[![Hugging Face Spaces](https://img.shields.io/badge/🤗%20Hugging%20Face-Spaces-blue)](https://huggingface.co/spaces/KenjieDec/RemBG)
[![Streamlit App](https://img.shields.io/badge/🎈%20Streamlit%20Community-Cloud-blue)](https://bgremoval.streamlit.app/)


Rembg is a tool to remove images background.

<p style="display: flex;align-items: center;justify-content: center;">
  <img src="https://raw.githubusercontent.com/danielgatis/rembg/master/examples/car-1.jpg" width="100" />
  <img src="https://raw.githubusercontent.com/danielgatis/rembg/master/examples/car-1.out.png" width="100" />
  <img src="https://raw.githubusercontent.com/danielgatis/rembg/master/examples/car-2.jpg" width="100" />
  <img src="https://raw.githubusercontent.com/danielgatis/rembg/master/examples/car-2.out.png" width="100" />
  <img src="https://raw.githubusercontent.com/danielgatis/rembg/master/examples/car-3.jpg" width="100" />
  <img src="https://raw.githubusercontent.com/danielgatis/rembg/master/examples/car-3.out.png" width="100" />
</p>

<p style="display: flex;align-items: center;justify-content: center;">
  <img src="https://raw.githubusercontent.com/danielgatis/rembg/master/examples/animal-1.jpg" width="100" />
  <img src="https://raw.githubusercontent.com/danielgatis/rembg/master/examples/animal-1.out.png" width="100" />
  <img src="https://raw.githubusercontent.com/danielgatis/rembg/master/examples/animal-2.jpg" width="100" />
  <img src="https://raw.githubusercontent.com/danielgatis/rembg/master/examples/animal-2.out.png" width="100" />
  <img src="https://raw.githubusercontent.com/danielgatis/rembg/master/examples/animal-3.jpg" width="100" />
  <img src="https://raw.githubusercontent.com/danielgatis/rembg/master/examples/animal-3.out.png" width="100" />
</p>

<p style="display: flex;align-items: center;justify-content: center;">
  <img src="https://raw.githubusercontent.com/danielgatis/rembg/master/examples/girl-1.jpg" width="100" />
  <img src="https://raw.githubusercontent.com/danielgatis/rembg/master/examples/girl-1.out.png" width="100" />
  <img src="https://raw.githubusercontent.com/danielgatis/rembg/master/examples/girl-2.jpg" width="100" />
  <img src="https://raw.githubusercontent.com/danielgatis/rembg/master/examples/girl-2.out.png" width="100" />
  <img src="https://raw.githubusercontent.com/danielgatis/rembg/master/examples/girl-3.jpg" width="100" />
  <img src="https://raw.githubusercontent.com/danielgatis/rembg/master/examples/girl-3.out.png" width="100" />
</p>

<p style="display: flex;align-items: center;justify-content: center;">
  <img src="https://raw.githubusercontent.com/danielgatis/rembg/master/examples/anime-girl-1.jpg" width="100" />
  <img src="https://raw.githubusercontent.com/danielgatis/rembg/master/examples/anime-girl-1.out.png" width="100" />
  <img src="https://raw.githubusercontent.com/danielgatis/rembg/master/examples/anime-girl-2.jpg" width="100" />
  <img src="https://raw.githubusercontent.com/danielgatis/rembg/master/examples/anime-girl-2.out.png" width="100" />
  <img src="https://raw.githubusercontent.com/danielgatis/rembg/master/examples/anime-girl-3.jpg" width="100" />
  <img src="https://raw.githubusercontent.com/danielgatis/rembg/master/examples/anime-girl-3.out.png" width="100" />
</p>

**If this project has helped you, please consider making a [donation](https://www.buymeacoffee.com/danielgatis).**

## Sponsor

<table>
  <tr>
    <td align="center" vertical-align="center">
      <a href="https://photoroom.com/api/remove-background?utm_source=rembg&utm_medium=github_webpage&utm_campaign=sponsor" >
        <img src="https://font-cdn.photoroom.com/media/api-logo.png" width="120px;" alt="Unsplash" />
      </a>
    </td>
    <td align="center" vertical-align="center">
      <b>PhotoRoom Remove Background API</b>
      <br />
      <a href="https://photoroom.com/api/remove-background?utm_source=rembg&utm_medium=github_webpage&utm_campaign=sponsor">https://photoroom.com/api</a>
      <br />
      <p width="200px">
        Fast and accurate background remover API<br/>
      </p>
    </td>
  </tr>
</table>

## Requirements

```
python: >3.7, <3.13
```

## Installation

CPU support:

```bash
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.

<p style="display: flex;align-items: center;justify-content: center;">
  <img src="https://raw.githubusercontent.com/danielgatis/rembg/master/onnxruntime-installation-matrix.png" width="400" />
</p>

If yes, just run:

```bash
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

```
SAM example

rembg i -m sam -x '{ "sam_prompt": [{"type": "point", "data": [724, 740], "label": 1}] }' examples/plants-1.jpg examples/plants-1.out.png
```

```
Custom model example

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.

```
rembg s --host 0.0.0.0 --port 7000 --log_level info
```

To see the complete endpoints documentation, go to: `http://localhost:7000/api`.

Remove the background from an image url

```
curl -s "http://localhost:7000/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:7000/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

```python
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

```python
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

```python
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

```python
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](USAGE.md) page.
## Usage as a docker

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

Try this:

```
docker run -v path/to/input:/rembg danielgatis/rembg i input.png path/to/output/output.png
```

## Models

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

The available models are:

-   u2net ([download](https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net.onnx), [source](https://github.com/xuebinqin/U-2-Net)): A pre-trained model for general use cases.
-   u2netp ([download](https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2netp.onnx), [source](https://github.com/xuebinqin/U-2-Net)): A lightweight version of u2net model.
-   u2net_human_seg ([download](https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net_human_seg.onnx), [source](https://github.com/xuebinqin/U-2-Net)): A pre-trained model for human segmentation.
-   u2net_cloth_seg ([download](https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net_cloth_seg.onnx), [source](https://github.com/levindabhi/cloth-segmentation)): 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](https://github.com/danielgatis/rembg/releases/download/v0.0.0/silueta.onnx), [source](https://github.com/xuebinqin/U-2-Net/issues/295)): Same as u2net but the size is reduced to 43Mb.
-   isnet-general-use ([download](https://github.com/danielgatis/rembg/releases/download/v0.0.0/isnet-general-use.onnx), [source](https://github.com/xuebinqin/DIS)): A new pre-trained model for general use cases.
-   isnet-anime ([download](https://github.com/danielgatis/rembg/releases/download/v0.0.0/isnet-anime.onnx), [source](https://github.com/SkyTNT/anime-segmentation)): A high-accuracy segmentation for anime character.
-   sam ([download encoder](https://github.com/danielgatis/rembg/releases/download/v0.0.0/vit_b-encoder-quant.onnx), [download decoder](https://github.com/danielgatis/rembg/releases/download/v0.0.0/vit_b-decoder-quant.onnx), [source](https://github.com/facebookresearch/segment-anything)): A pre-trained model for any use cases.

### How to train your own model

If You need more fine tuned models try this:
https://github.com/danielgatis/rembg/issues/193#issuecomment-1055534289


## Some video tutorials

- https://www.youtube.com/watch?v=3xqwpXjxyMQ
- https://www.youtube.com/watch?v=dFKRGXdkGJU
- https://www.youtube.com/watch?v=Ai-BS_T7yjE
- https://www.youtube.com/watch?v=D7W-C0urVcQ

## References

- https://arxiv.org/pdf/2005.09007.pdf
- https://github.com/NathanUA/U-2-Net
- https://github.com/pymatting/pymatting

## FAQ

### When will this library provide support for Python version 3.xx?

This library directly depends on the [onnxruntime](https://pypi.org/project/onnxruntime) library. Therefore, we can only update the Python version when [onnxruntime](https://pypi.org/project/onnxruntime) provides support for that specific version.

## Buy me a coffee

Liked some of my work? Buy me a coffee (or more likely a beer)

<a href="https://www.buymeacoffee.com/danielgatis" target="_blank"><img src="https://bmc-cdn.nyc3.digitaloceanspaces.com/BMC-button-images/custom_images/orange_img.png" alt="Buy Me A Coffee" style="height: auto !important;width: auto !important;"></a>

## License

Copyright (c) 2020-present [Daniel Gatis](https://github.com/danielgatis)

Licensed under [MIT License](./LICENSE.txt)

            

Raw data

            {
    "_id": null,
    "home_page": "https://github.com/danielgatis/rembg",
    "name": "rembg",
    "maintainer": "",
    "docs_url": null,
    "requires_python": ">=3.8, <3.13",
    "maintainer_email": "",
    "keywords": "remove,background,u2net",
    "author": "Daniel Gatis",
    "author_email": "danielgatis@gmail.com",
    "download_url": "https://files.pythonhosted.org/packages/42/fe/231ddd6963568dd98b507eaf4b2f12f37585fc15c316bcb9f7f8c1faba3e/rembg-2.0.56.tar.gz",
    "platform": null,
    "description": "# Rembg\n\n[![Downloads](https://img.shields.io/pypi/dm/rembg.svg)](https://img.shields.io/pypi/dm/rembg.svg)\n[![License](https://img.shields.io/badge/License-MIT-blue.svg)](https://img.shields.io/badge/License-MIT-blue.svg)\n[![Hugging Face Spaces](https://img.shields.io/badge/\ud83e\udd17%20Hugging%20Face-Spaces-blue)](https://huggingface.co/spaces/KenjieDec/RemBG)\n[![Streamlit App](https://img.shields.io/badge/\ud83c\udf88%20Streamlit%20Community-Cloud-blue)](https://bgremoval.streamlit.app/)\n\n\nRembg is a tool to remove images background.\n\n<p style=\"display: flex;align-items: center;justify-content: center;\">\n  <img src=\"https://raw.githubusercontent.com/danielgatis/rembg/master/examples/car-1.jpg\" width=\"100\" />\n  <img src=\"https://raw.githubusercontent.com/danielgatis/rembg/master/examples/car-1.out.png\" width=\"100\" />\n  <img src=\"https://raw.githubusercontent.com/danielgatis/rembg/master/examples/car-2.jpg\" width=\"100\" />\n  <img src=\"https://raw.githubusercontent.com/danielgatis/rembg/master/examples/car-2.out.png\" width=\"100\" />\n  <img src=\"https://raw.githubusercontent.com/danielgatis/rembg/master/examples/car-3.jpg\" width=\"100\" />\n  <img src=\"https://raw.githubusercontent.com/danielgatis/rembg/master/examples/car-3.out.png\" width=\"100\" />\n</p>\n\n<p style=\"display: flex;align-items: center;justify-content: center;\">\n  <img src=\"https://raw.githubusercontent.com/danielgatis/rembg/master/examples/animal-1.jpg\" width=\"100\" />\n  <img src=\"https://raw.githubusercontent.com/danielgatis/rembg/master/examples/animal-1.out.png\" width=\"100\" />\n  <img src=\"https://raw.githubusercontent.com/danielgatis/rembg/master/examples/animal-2.jpg\" width=\"100\" />\n  <img src=\"https://raw.githubusercontent.com/danielgatis/rembg/master/examples/animal-2.out.png\" width=\"100\" />\n  <img src=\"https://raw.githubusercontent.com/danielgatis/rembg/master/examples/animal-3.jpg\" width=\"100\" />\n  <img src=\"https://raw.githubusercontent.com/danielgatis/rembg/master/examples/animal-3.out.png\" width=\"100\" />\n</p>\n\n<p style=\"display: flex;align-items: center;justify-content: center;\">\n  <img src=\"https://raw.githubusercontent.com/danielgatis/rembg/master/examples/girl-1.jpg\" width=\"100\" />\n  <img src=\"https://raw.githubusercontent.com/danielgatis/rembg/master/examples/girl-1.out.png\" width=\"100\" />\n  <img src=\"https://raw.githubusercontent.com/danielgatis/rembg/master/examples/girl-2.jpg\" width=\"100\" />\n  <img src=\"https://raw.githubusercontent.com/danielgatis/rembg/master/examples/girl-2.out.png\" width=\"100\" />\n  <img src=\"https://raw.githubusercontent.com/danielgatis/rembg/master/examples/girl-3.jpg\" width=\"100\" />\n  <img src=\"https://raw.githubusercontent.com/danielgatis/rembg/master/examples/girl-3.out.png\" width=\"100\" />\n</p>\n\n<p style=\"display: flex;align-items: center;justify-content: center;\">\n  <img src=\"https://raw.githubusercontent.com/danielgatis/rembg/master/examples/anime-girl-1.jpg\" width=\"100\" />\n  <img src=\"https://raw.githubusercontent.com/danielgatis/rembg/master/examples/anime-girl-1.out.png\" width=\"100\" />\n  <img src=\"https://raw.githubusercontent.com/danielgatis/rembg/master/examples/anime-girl-2.jpg\" width=\"100\" />\n  <img src=\"https://raw.githubusercontent.com/danielgatis/rembg/master/examples/anime-girl-2.out.png\" width=\"100\" />\n  <img src=\"https://raw.githubusercontent.com/danielgatis/rembg/master/examples/anime-girl-3.jpg\" width=\"100\" />\n  <img src=\"https://raw.githubusercontent.com/danielgatis/rembg/master/examples/anime-girl-3.out.png\" width=\"100\" />\n</p>\n\n**If this project has helped you, please consider making a [donation](https://www.buymeacoffee.com/danielgatis).**\n\n## Sponsor\n\n<table>\n  <tr>\n    <td align=\"center\" vertical-align=\"center\">\n      <a href=\"https://photoroom.com/api/remove-background?utm_source=rembg&utm_medium=github_webpage&utm_campaign=sponsor\" >\n        <img src=\"https://font-cdn.photoroom.com/media/api-logo.png\" width=\"120px;\" alt=\"Unsplash\" />\n      </a>\n    </td>\n    <td align=\"center\" vertical-align=\"center\">\n      <b>PhotoRoom Remove Background API</b>\n      <br />\n      <a href=\"https://photoroom.com/api/remove-background?utm_source=rembg&utm_medium=github_webpage&utm_campaign=sponsor\">https://photoroom.com/api</a>\n      <br />\n      <p width=\"200px\">\n        Fast and accurate background remover API<br/>\n      </p>\n    </td>\n  </tr>\n</table>\n\n## Requirements\n\n```\npython: >3.7, <3.13\n```\n\n## Installation\n\nCPU support:\n\n```bash\npip install rembg # for library\npip install rembg[cli] # for library + cli\n```\n\nGPU support:\n\nFirst of all, you need to check if your system supports the `onnxruntime-gpu`.\n\nGo to https://onnxruntime.ai and check the installation matrix.\n\n<p style=\"display: flex;align-items: center;justify-content: center;\">\n  <img src=\"https://raw.githubusercontent.com/danielgatis/rembg/master/onnxruntime-installation-matrix.png\" width=\"400\" />\n</p>\n\nIf yes, just run:\n\n```bash\npip install rembg[gpu] # for library\npip install rembg[gpu,cli] # for library + cli\n```\n\n## Usage as a cli\n\nAfter the installation step you can use rembg just typing `rembg` in your terminal window.\n\nThe `rembg` command has 4 subcommands, one for each input type:\n- `i` for files\n- `p` for folders\n- `s` for http server\n- `b` for RGB24 pixel binary stream\n\nYou can get help about the main command using:\n\n```\nrembg --help\n```\n\nAs well, about all the subcommands using:\n\n```\nrembg <COMMAND> --help\n```\n\n### rembg `i`\n\nUsed when input and output are files.\n\nRemove the background from a remote image\n\n```\ncurl -s http://input.png | rembg i > output.png\n```\n\nRemove the background from a local file\n\n```\nrembg i path/to/input.png path/to/output.png\n```\n\nRemove the background specifying a model\n\n```\nrembg i -m u2netp path/to/input.png path/to/output.png\n```\n\nRemove the background returning only the mask\n\n```\nrembg i -om path/to/input.png path/to/output.png\n```\n\n\nRemove the background applying an alpha matting\n\n```\nrembg i -a path/to/input.png path/to/output.png\n```\n\nPassing extras parameters\n\n```\nSAM example\n\nrembg i -m sam -x '{ \"sam_prompt\": [{\"type\": \"point\", \"data\": [724, 740], \"label\": 1}] }' examples/plants-1.jpg examples/plants-1.out.png\n```\n\n```\nCustom model example\n\nrembg i -m u2net_custom -x '{\"model_path\": \"~/.u2net/u2net.onnx\"}' path/to/input.png path/to/output.png\n```\n\n### rembg `p`\n\nUsed when input and output are folders.\n\nRemove the background from all images in a folder\n\n```\nrembg p path/to/input path/to/output\n```\n\nSame as before, but watching for new/changed files to process\n\n```\nrembg p -w path/to/input path/to/output\n```\n\n### rembg `s`\n\nUsed to start http server.\n\n```\nrembg s --host 0.0.0.0 --port 7000 --log_level info\n```\n\nTo see the complete endpoints documentation, go to: `http://localhost:7000/api`.\n\nRemove the background from an image url\n\n```\ncurl -s \"http://localhost:7000/api/remove?url=http://input.png\" -o output.png\n```\n\nRemove the background from an uploaded image\n\n```\ncurl -s -F file=@/path/to/input.jpg \"http://localhost:7000/api/remove\"  -o output.png\n```\n\n### rembg `b`\n\nProcess 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.\n\n```\nrembg b image_width image_height -o output_specifier\n```\n\nArguments:\n\n- image_width : width of input image(s)\n- image_height : height of input image(s)\n- 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.\n\nExample usage with FFMPEG:\n\n```\nffmpeg -i input.mp4 -ss 10 -an -f rawvideo -pix_fmt rgb24 pipe:1 | rembg b 1280 720 -o folder/output-%03u.png\n```\n\nThe 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.\n\n\n## Usage as a library\n\nInput and output as bytes\n\n```python\nfrom rembg import remove\n\ninput_path = 'input.png'\noutput_path = 'output.png'\n\nwith open(input_path, 'rb') as i:\n    with open(output_path, 'wb') as o:\n        input = i.read()\n        output = remove(input)\n        o.write(output)\n```\n\nInput and output as a PIL image\n\n```python\nfrom rembg import remove\nfrom PIL import Image\n\ninput_path = 'input.png'\noutput_path = 'output.png'\n\ninput = Image.open(input_path)\noutput = remove(input)\noutput.save(output_path)\n```\n\nInput and output as a numpy array\n\n```python\nfrom rembg import remove\nimport cv2\n\ninput_path = 'input.png'\noutput_path = 'output.png'\n\ninput = cv2.imread(input_path)\noutput = remove(input)\ncv2.imwrite(output_path, output)\n```\n\nHow to iterate over files in a performatic way\n\n```python\nfrom pathlib import Path\nfrom rembg import remove, new_session\n\nsession = new_session()\n\nfor file in Path('path/to/folder').glob('*.png'):\n    input_path = str(file)\n    output_path = str(file.parent / (file.stem + \".out.png\"))\n\n    with open(input_path, 'rb') as i:\n        with open(output_path, 'wb') as o:\n            input = i.read()\n            output = remove(input, session=session)\n            o.write(output)\n```\nTo see a full list of examples on how to use rembg, go to the [examples](USAGE.md) page.\n## Usage as a docker\n\nJust replace the `rembg` command for `docker run danielgatis/rembg`.\n\nTry this:\n\n```\ndocker run -v path/to/input:/rembg danielgatis/rembg i input.png path/to/output/output.png\n```\n\n## Models\n\nAll models are downloaded and saved in the user home folder in the `.u2net` directory.\n\nThe available models are:\n\n-   u2net ([download](https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net.onnx), [source](https://github.com/xuebinqin/U-2-Net)): A pre-trained model for general use cases.\n-   u2netp ([download](https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2netp.onnx), [source](https://github.com/xuebinqin/U-2-Net)): A lightweight version of u2net model.\n-   u2net_human_seg ([download](https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net_human_seg.onnx), [source](https://github.com/xuebinqin/U-2-Net)): A pre-trained model for human segmentation.\n-   u2net_cloth_seg ([download](https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net_cloth_seg.onnx), [source](https://github.com/levindabhi/cloth-segmentation)): A pre-trained model for Cloths Parsing from human portrait. Here clothes are parsed into 3 category: Upper body, Lower body and Full body.\n-   silueta ([download](https://github.com/danielgatis/rembg/releases/download/v0.0.0/silueta.onnx), [source](https://github.com/xuebinqin/U-2-Net/issues/295)): Same as u2net but the size is reduced to 43Mb.\n-   isnet-general-use ([download](https://github.com/danielgatis/rembg/releases/download/v0.0.0/isnet-general-use.onnx), [source](https://github.com/xuebinqin/DIS)): A new pre-trained model for general use cases.\n-   isnet-anime ([download](https://github.com/danielgatis/rembg/releases/download/v0.0.0/isnet-anime.onnx), [source](https://github.com/SkyTNT/anime-segmentation)): A high-accuracy segmentation for anime character.\n-   sam ([download encoder](https://github.com/danielgatis/rembg/releases/download/v0.0.0/vit_b-encoder-quant.onnx), [download decoder](https://github.com/danielgatis/rembg/releases/download/v0.0.0/vit_b-decoder-quant.onnx), [source](https://github.com/facebookresearch/segment-anything)): A pre-trained model for any use cases.\n\n### How to train your own model\n\nIf You need more fine tuned models try this:\nhttps://github.com/danielgatis/rembg/issues/193#issuecomment-1055534289\n\n\n## Some video tutorials\n\n- https://www.youtube.com/watch?v=3xqwpXjxyMQ\n- https://www.youtube.com/watch?v=dFKRGXdkGJU\n- https://www.youtube.com/watch?v=Ai-BS_T7yjE\n- https://www.youtube.com/watch?v=D7W-C0urVcQ\n\n## References\n\n- https://arxiv.org/pdf/2005.09007.pdf\n- https://github.com/NathanUA/U-2-Net\n- https://github.com/pymatting/pymatting\n\n## FAQ\n\n### When will this library provide support for Python version 3.xx?\n\nThis library directly depends on the [onnxruntime](https://pypi.org/project/onnxruntime) library. Therefore, we can only update the Python version when [onnxruntime](https://pypi.org/project/onnxruntime) provides support for that specific version.\n\n## Buy me a coffee\n\nLiked some of my work? Buy me a coffee (or more likely a beer)\n\n<a href=\"https://www.buymeacoffee.com/danielgatis\" target=\"_blank\"><img src=\"https://bmc-cdn.nyc3.digitaloceanspaces.com/BMC-button-images/custom_images/orange_img.png\" alt=\"Buy Me A Coffee\" style=\"height: auto !important;width: auto !important;\"></a>\n\n## License\n\nCopyright (c) 2020-present [Daniel Gatis](https://github.com/danielgatis)\n\nLicensed under [MIT License](./LICENSE.txt)\n",
    "bugtrack_url": null,
    "license": "",
    "summary": "Remove image background",
    "version": "2.0.56",
    "project_urls": {
        "Homepage": "https://github.com/danielgatis/rembg"
    },
    "split_keywords": [
        "remove",
        "background",
        "u2net"
    ],
    "urls": [
        {
            "comment_text": "",
            "digests": {
                "blake2b_256": "85a9fbb32cb436815a07b0e066503e42a9a259025843f825cb08d9abcc7ae228",
                "md5": "5767b7bfc2532f499a7e9142019e7081",
                "sha256": "5f2a2136969f1e46771951f0318e11a0cbb77ea553562392e7a2d48cda125ab9"
            },
            "downloads": -1,
            "filename": "rembg-2.0.56-py3-none-any.whl",
            "has_sig": false,
            "md5_digest": "5767b7bfc2532f499a7e9142019e7081",
            "packagetype": "bdist_wheel",
            "python_version": "py3",
            "requires_python": ">=3.8, <3.13",
            "size": 32890,
            "upload_time": "2024-03-12T22:54:57",
            "upload_time_iso_8601": "2024-03-12T22:54:57.260818Z",
            "url": "https://files.pythonhosted.org/packages/85/a9/fbb32cb436815a07b0e066503e42a9a259025843f825cb08d9abcc7ae228/rembg-2.0.56-py3-none-any.whl",
            "yanked": false,
            "yanked_reason": null
        },
        {
            "comment_text": "",
            "digests": {
                "blake2b_256": "42fe231ddd6963568dd98b507eaf4b2f12f37585fc15c316bcb9f7f8c1faba3e",
                "md5": "d5eb2fc2732f56a4d52b8e13db734682",
                "sha256": "2ca8aa0460674fd93622de5a95f1e8329c87191972599a99b9989b8d71686e1f"
            },
            "downloads": -1,
            "filename": "rembg-2.0.56.tar.gz",
            "has_sig": false,
            "md5_digest": "d5eb2fc2732f56a4d52b8e13db734682",
            "packagetype": "sdist",
            "python_version": "source",
            "requires_python": ">=3.8, <3.13",
            "size": 44183,
            "upload_time": "2024-03-12T22:54:58",
            "upload_time_iso_8601": "2024-03-12T22:54:58.855809Z",
            "url": "https://files.pythonhosted.org/packages/42/fe/231ddd6963568dd98b507eaf4b2f12f37585fc15c316bcb9f7f8c1faba3e/rembg-2.0.56.tar.gz",
            "yanked": false,
            "yanked_reason": null
        }
    ],
    "upload_time": "2024-03-12 22:54:58",
    "github": true,
    "gitlab": false,
    "bitbucket": false,
    "codeberg": false,
    "github_user": "danielgatis",
    "github_project": "rembg",
    "travis_ci": false,
    "coveralls": false,
    "github_actions": true,
    "lcname": "rembg"
}
        
Elapsed time: 0.20144s