Smooth edges

app.py

@@ -202,7 +202,7 @@ def smooth_and_denoise(mask):
 
 def smooth_edges(mask):
     """
-    Smooth edges of a binary mask using morphological operations.
+    Smooth edges of a binary mask using morphological operations and anti-aliasing resizing.
     :param mask: The binary mask of the signature.
     :return: Mask with smoothed edges.
     """
@@ -221,7 +221,19 @@ def smooth_edges(mask):
     # Dilate the mask to make the signature slightly thicker
     dilated = cv2.dilate(opening, kernel, iterations=1)
     
-    return dilated
+    # Convert dilated mask to a PIL image for anti-aliasing resizing
+    pil_mask = Image.fromarray(dilated)
+    
+    # Resize the image to a smaller size, then back to the original size
+    small_size = (pil_mask.width // 2, pil_mask.height // 2)
+    pil_mask_small = pil_mask.resize(small_size, Image.ANTIALIAS)
+    pil_mask_smooth = pil_mask_small.resize(pil_mask.size, Image.ANTIALIAS)
+    
+    # Convert back to a numpy array
+    smoothed_mask = np.array(pil_mask_smooth)
+    final_mask = cv2.bilateralFilter(mask, d=9, sigmaColor=75, sigmaSpace=75)
+
+    return final_mask
 
     
 def predict(net,  inputs_val, shapes_val, hypar, device):