Update model/cloth_masker.py

model/cloth_masker.py

@@ -244,8 +244,15 @@ class AutoMasker:
 
 
         mask_area = (np.ones_like(densepose_mask) & (~weak_protect_area) & (~background_area)) | mask_dense_area
+        mask_area = hull_mask(mask_area * 255) // 255  
+        mask_area = mask_area & (~weak_protect_area)
+        mask_area = cv2.GaussianBlur(mask_area * 255, (kernal_size, kernal_size), 0)
+        mask_area[mask_area < 25] = 0
+        mask_area[mask_area >= 25] = 1
+        
         mask_area = (mask_area | strong_mask_area) & (~strong_protect_area) 
-
+        mask_area = cv2.dilate(mask_area, dilate_kernel, iterations=1)
+        
         return Image.fromarray(mask_area * 255)
         
     def __call__(