LucindeAI/rbg

Rbg is a simple flask website to remove images background by Rembg.

Rbg

Rbg is a simple flask website to remove images background by Rembg.

Project Tree

.
├── app.py
├── LICENSE
├── __pycache__
│   └── app.cpython-310.pyc
├── README.md
├── requirements.txt
├── templates
│   └── index.html
├── .u2net                   # Download models to this folder by yourself
└── uploads                  # folder as `UPLOAD_FOLDER` folder in app.py

Installation

pip install -r requirements.txt

Models

You should download models to the .u2net folder by yourself.

The available models are:

• u2net (download - alternative, source): A pre-trained model for general use cases.
• u2netp (download - alternative, source): A lightweight version of u2net model.
• u2net_human_seg (download - alternative, source): A pre-trained model for human segmentation.
• u2net_cloth_seg (download - alternative, 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.

Launch

flask run