feat: add inpaint mask generation functionality

uses rembg, see https://github.com/danielgatis/rembg
This commit is contained in:
Manuel Schmid
2024-01-24 02:22:50 +01:00
parent f6d67d7655
commit db5512fb93
5 changed files with 44 additions and 3 deletions
+15
View File
@@ -0,0 +1,15 @@
from rembg import remove, new_session
def generate_mask_from_image(image, mask_model):
if image is None:
return
if 'image' in image:
image = image['image']
return remove(
image,
session=new_session(mask_model),
only_mask=True
)