Rembg integration

[graemeniedermayer]

The pull request integrates u2-net background removal. So you can remove the background before or after the depth is calculated. It allows you to save the background removed image, and a bk/white mask. default is set to off.

There's one line that might cause trouble on Windows
os.environ["U2NET_HOME"] = str(bg_model_dir)
this line tells rembg where to look for the model.

I haven't fully tested with 3dinpaint or stereo images. An alpha channel is added if using pre-depth calculate which might cause issues with them.

Removing the background before calculating depth for certain images can really improve the details.

depth with background removed

depth with normal background

orginal

background removed

mask (this has an alpha channel if you using light ui this will look white)

There's also u2-net clothing segmentation, but that seems a little out of scope for the moment.

[thygate]

Hey, great! Looks like a clean integration, I've not tested it yet, does it work on Windows ?

[graemeniedermayer]

I don't have a Windows machine. It should work. I just noticed there was quite a few stackoverflow questions about Windows and os.environ , but most are related to permanently setting environment variable (which is not required here).

It should still work without the line. It would just download the models to ~/.u2net folder instead of the webui's /model folder

[thygate]

Hey sorry for the delay, I've been a bit preoccupied .. Will merge asap.