Merge branch 'feature/add-inpaint-mask-generation'

2024-01-27 21:48:41 +01:00 · 2024-01-27 21:48:41 +01:00 · 9685a90090
parent 7185abb8ba f97e682e80
commit 9685a90090
1 changed files with 1 additions and 0 deletions
--- a/extras/inpaint_mask.py
+++ b/extras/inpaint_mask.py
@ -29,6 +29,7 @@ def generate_mask_from_image(image, mask_model, extras):

    if mask_model == 'sam':
        boxes = run_grounded_sam(Image.fromarray(image), extras['sam_prompt_text'], box_threshold=extras['box_threshold'], text_threshold=extras['text_threshold'])
+        boxes = np.array([[0, 0, image.shape[1], image.shape[0]]]) if len(boxes) == 0 else boxes
        extras['sam_prompt'] = []
        for idx, box in enumerate(boxes):
            extras['sam_prompt'] += [{"type": "rectangle", "data": box.tolist()}]