search

chaiNNer-org / spandrel

Spandrel gives your project support for various PyTorch architectures meant for AI Super-Resolution, restoration, and inpainting. Based on the model support implemented in chaiNNer.
MIT License
103 stars 7 forks source link

add remove background models in PyTorch #150

Open wolfkingal2000 opened 4 months ago

wolfkingal2000 commented 4 months ago

add remove background like this models https://huggingface.co/briaai/RMBG-1.4 https://github.com/danielgatis/rembg https://github.com/SkyTNT/anime-segmentation make can use this models in software tanks

RunDevelopment commented 4 months ago

https://github.com/danielgatis/rembg

We already support that. See ONNX Remove Background.

joeyballentine commented 4 months ago

I still think it would be worth adding pytorch versions of rembg and potentially others as well

wolfkingal2000 commented 4 months ago

https://github.com/danielgatis/rembg

We already support that. See ONNX Remove Background.

Yes, i use onnx vr, but I dont work my pc. i want to use pytorch vr

martirsadota commented 3 months ago

To add, I know of a bunch more background removal models:

InSPyReNet, the previous SOTA in Papers With Code: https://github.com/plemeri/InSPyReNet BiRefNet, the current SOTA: https://github.com/zhengpeng7/birefnet and TRACER: https://github.com/Karel911/TRACER