kos94ok / bgremoverapp Goto Github PK

Hi,
How can I use it for doing image processing for the output without the backround, and then return the original backround after my changes on the main object?
Do you have code example for this?

Thank

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')