fix hair expand

app.py

@@ -60,6 +60,7 @@ def get_hair_mask(category_mask_np, should_dilate=False):
 
     face_indices = np.where(face_skin_mask)
     min_face_y = np.min(face_indices[0])
+    max_face_y = np.max(face_indices[0])
 
     labeled_hair, hair_features = label(hair_mask)
     top_hair_mask = np.zeros_like(hair_mask)
@@ -70,11 +71,10 @@ def get_hair_mask(category_mask_np, should_dilate=False):
         if min_component_y <= min_face_y:
             top_hair_mask[component_mask] = True
     
-    expanded_face_mask = binary_dilation(face_skin_mask, iterations=40)
     # Combine the reference masks (body, clothes)
     reference_mask = np.logical_or(body_skin_mask, clothes_mask)
-    # Exclude the expanded face mask from the reference mask
-    reference_mask = np.logical_and(reference_mask, ~expanded_face_mask)
+    # Remove the area above the face by 40 pixels
+    reference_mask[:max_face_y+40, :] = 0
 
     # Expand the hair mask downward until it reaches the reference areas
     expanded_hair_mask = top_hair_mask
@@ -84,8 +84,8 @@ def get_hair_mask(category_mask_np, should_dilate=False):
     # Trim the expanded_hair_mask
     # 1. Remove the area above hair_mask by 20 pixels
     hair_indices = np.where(hair_mask)
-    min_hair_y = np.min(hair_indices[0]) - 20
-    expanded_hair_mask[:min_hair_y, :] = 0
+    min_hair_y = np.min(hair_indices[0])
+    expanded_hair_mask[:min_hair_y - 20, :] = 0
 
     # 2. Remove the areas on both sides that exceed the clothing coordinates
     clothes_indices = np.where(clothes_mask)