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

feat: only process sam_options.dino_erode_or_dilate when set

Browse Source
This commit is contained in:
Manuel Schmid
2024-06-10 23:47:09 +02:00
parent ab3f4df935
commit b5b33361c9
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
extras/inpaint_mask.py
+1
View File
@@ -83,6 +83,7 @@ def generate_mask_from_image(image: np.ndarray, mask_model: str = 'sam', extras=
if boxes.size(0) > 0: if boxes.size(0) > 0:
sam_predictor.set_image(image) sam_predictor.set_image(image)
if sam_options.dino_erode_or_dilate != 0:
for index in range(boxes.size(0)): for index in range(boxes.size(0)):
assert boxes.size(1) == 4 assert boxes.size(1) == 4
boxes[index][0] -= sam_options.dino_erode_or_dilate boxes[index][0] -= sam_options.dino_erode_or_dilate