add negative padding

processing/mask.py

@@ -3,6 +3,7 @@ import numpy as np
 import torch
 from PIL import Image, ImageFilter
 from rembg import remove
+from scipy.ndimage import binary_dilation
 from torchvision.transforms import Compose
 
 from DPT.dpt.transforms import PrepareForNet, NormalizeImage, Resize
@@ -13,7 +14,10 @@ def create_mask(image, blur=0, padding=0):
     rm_bg = Image.fromarray((rm_bg * 255).astype(np.uint8))
     rm_bg = rm_bg.resize(image.size, resample=Image.BILINEAR)
     # Create a padding of 5 pixels around the object in the mask
-    padded_mask = np.pad(rm_bg, pad_width=padding, mode='constant', constant_values=0)
+    if padding > 0:
+        padded_mask = np.pad(rm_bg, pad_width=padding, mode='constant', constant_values=0)
+    else:
+        padded_mask = binary_dilation(np.array(rm_bg), iterations=-padding)
 
     # Convert mask back to uint8 for PIL compatibility
     pil_mask = Image.fromarray((padded_mask * 255).astype(np.uint8))

requirements.txt

@@ -19,4 +19,5 @@ matplotlib
 setuptools
 safetensors
 huggingface-hub
+scipy
 ultralytics~=8.2.48