Tool for remove background | U2NET, TRACER

Quick Start

Tested on PyTorch 1.11.0
Check out this folder. We have built a pip package that can remove background an input image with one line of code.

Install with

pip install bgRemoverApp

Code demo

from PIL import Image
import bgRemoverApp

model_name = "U2NET" # U2NET, TRACER
model_path = './ckpt/u2net.pth' # './ckpt/TRACER-Efficient-7.pth'
input_image = Image.open('./input.jpg')

model_pred = bgRemoverApp.load_model(model_name, model_path)
output_image = bgRemoverApp.remove_bg(input_image, model_name, model_pred)
output_image.save('output.png')