ZHKKKe / MODNet

A Trimap-Free Portrait Matting Solution in Real Time [AAAI 2022]
Apache License 2.0
3.67k stars 619 forks source link

Improving quality of background removal #159

Open aksthelion opened 2 years ago

aksthelion commented 2 years ago

I've tried this script to test background removal and compared it to commercially available AI powered solution and found that this model is not as good as commercial ones. If you zoom in image and compare you will see that this model doesn't produce smooth edges.

Can it be improved? Is there any further plan to improve it?

manzke commented 2 years ago

you can use pymatting to improve it. example https://github.com/danielgatis/rembg

ZHKKKe commented 2 years ago

@aksthelion Thanks for your attention. For commercial model, like remove.bg, their labeling quality is higher an their label number is larger. They also use the technique like foreground estimation (as mentioned by @manzke above) to improve the boundaries. As our code is just for research purpose, we do not include these technologies.