Add options to u2net_cloth_seg

This commit is contained in:
rayronvictor
2024-01-24 21:26:23 -03:00
parent dea4b6c15c
commit f0fb9783e1
4 changed files with 51 additions and 23 deletions
+3 -2
View File
@@ -1,7 +1,7 @@
from rembg import remove, new_session
def generate_mask_from_image(image, mask_model):
def generate_mask_from_image(image, mask_model, extras):
if image is None:
return
@@ -11,5 +11,6 @@ def generate_mask_from_image(image, mask_model):
return remove(
image,
session=new_session(mask_model),
only_mask=True
only_mask=True,
**extras
)