From b5b33361c96c7ba8d845264408798b91def37172 Mon Sep 17 00:00:00 2001 From: Manuel Schmid Date: Mon, 10 Jun 2024 23:47:09 +0200 Subject: [PATCH] feat: only process sam_options.dino_erode_or_dilate when set --- extras/inpaint_mask.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/extras/inpaint_mask.py b/extras/inpaint_mask.py index a4f0e7c4..7bb671f5 100644 --- a/extras/inpaint_mask.py +++ b/extras/inpaint_mask.py @@ -83,12 +83,13 @@ def generate_mask_from_image(image: np.ndarray, mask_model: str = 'sam', extras= if boxes.size(0) > 0: sam_predictor.set_image(image) - for index in range(boxes.size(0)): - assert boxes.size(1) == 4 - boxes[index][0] -= sam_options.dino_erode_or_dilate - boxes[index][1] -= sam_options.dino_erode_or_dilate - boxes[index][2] += sam_options.dino_erode_or_dilate - boxes[index][3] += sam_options.dino_erode_or_dilate + if sam_options.dino_erode_or_dilate != 0: + for index in range(boxes.size(0)): + assert boxes.size(1) == 4 + boxes[index][0] -= sam_options.dino_erode_or_dilate + boxes[index][1] -= sam_options.dino_erode_or_dilate + boxes[index][2] += sam_options.dino_erode_or_dilate + boxes[index][3] += sam_options.dino_erode_or_dilate if sam_options.dino_debug: from PIL import ImageDraw, Image