let5see/IOPaint
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

RemoveBg plugin add birefnet models, require rembg>=2.0.59

Browse Source
This commit is contained in:
Qing
2024-11-01 18:28:04 +08:00
parent b4b49bb88e
commit 2098d687d1
3 changed files with 25 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
iopaint/schema.py
View File

@@ -144,13 +144,21 @@ class RealESRGANModel(Choices):
class RemoveBGModel(Choices):
briaai_rmbg_1_4 = "briaai/RMBG-1.4"
# models from https://github.com/danielgatis/rembg
u2net = "u2net"
u2netp = "u2netp"
u2net_human_seg = "u2net_human_seg"
u2net_cloth_seg = "u2net_cloth_seg"
silueta = "silueta"
isnet_general_use = "isnet-general-use"
briaai_rmbg_1_4 = "briaai/RMBG-1.4"
birefnet_general = "birefnet-general"
birefnet_general_lite = "birefnet-general-lite"
birefnet_portrait = "birefnet-portrait"
birefnet_dis = "birefnet-dis"
birefnet_hrsod = "birefnet-hrsod"
birefnet_cod = "birefnet-cod"
birefnet_massive = "birefnet-massive"
class Device(Choices):