Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix error when run_grounded_sam find no boxes

Browse Source
This commit is contained in:
rayronvictor
2024-01-26 07:58:57 -03:00
parent dd2fd04fd7
commit dd5264dd14
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
extras/inpaint_mask.py
+1
View File
@@ -29,6 +29,7 @@ def generate_mask_from_image(image, mask_model, extras):
if mask_model == 'sam': 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 = 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'] = [] extras['sam_prompt'] = []
for idx, box in enumerate(boxes): for idx, box in enumerate(boxes):
extras['sam_prompt'] += [{"type": "rectangle", "data": box.tolist()}] extras['sam_prompt'] += [{"type": "rectangle", "data": box.tolist()}]