mirror of
https://github.com/lllyasviel/Fooocus.git
synced 2026-08-16 13:13:16 +02:00
Merge branch 'feature/add-inpaint-mask-generation'
# Conflicts: # modules/config.py # requirements_versions.txt
This commit is contained in:
@@ -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
|
||||
)
|
||||
Reference in New Issue
Block a user