Update utils_mask.py

utils_mask.py

@@ -89,6 +89,10 @@ def get_mask_location(model_type, category, model_parse: Image.Image, keypoint:
         parse_mask += (parse_array == 12).astype(np.float32) + \
                       (parse_array == 13).astype(np.float32)
 
+        # Fill gaps between legs
+        leg_mask = cv2.dilate(parse_mask.astype(np.uint8), np.ones((5, 5), np.uint8), iterations=5)
+        parse_mask += leg_mask
+
     elif category == 'upper_body':
         parse_mask = (parse_array == 4).astype(np.float32) + (parse_array == 7).astype(np.float32)
         parser_mask_fixed_lower_cloth = (parse_array == label_map["skirt"]).astype(np.float32) + \