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

refactor: add info to inpaint_mask_sam_prompt_text, rename variable for dino debugging

Browse Source
This commit is contained in:
Manuel Schmid
2024-06-09 23:09:46 +02:00
parent 57c049858c
commit 09e23f5509
3 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
extras/inpaint_mask.py
+3 -3
View File
@@ -33,11 +33,11 @@ def generate_mask_from_image(image: np.ndarray, mask_model: str, extras: dict, b
if debug_dino:
from PIL import ImageDraw, Image
image_with_boxes = Image.new("RGB", (image.shape[1], image.shape[0]), color="black")
draw = ImageDraw.Draw(image_with_boxes)
debug_dino_image = Image.new("RGB", (image.shape[1], image.shape[0]), color="black")
draw = ImageDraw.Draw(debug_dino_image)
for box in extras['sam_prompt']:
draw.rectangle(box['data'], fill="white")
return np.array(image_with_boxes)
return np.array(debug_dino_image)
return remove(
image,