patrickligardes commited on
Commit
b7a4430
ยท
verified ยท
1 Parent(s): 1255c14

Update utils_mask.py

Browse files
Files changed (1) hide show
  1. utils_mask.py +2 -3
utils_mask.py CHANGED
@@ -88,9 +88,8 @@ def get_mask_location(model_type, category, model_parse: Image.Image, keypoint:
88
  # Create a separate mask for the legs
89
  parse_mask_legs = (parse_array == 12).astype(np.float32) + \
90
  (parse_array == 13).astype(np.float32) + \
91
- (parse_array == 6).astype(np.float32)
92
-
93
- parser_mask_changeable += np.logical_and(parse_array, np.logical_not(parser_mask_fixed))
94
 
95
  # Fill gaps between legs
96
  leg_mask_filled = cv2.dilate(parse_mask_legs.astype(np.uint8), np.ones((6, 6), np.uint8), iterations=6)
 
88
  # Create a separate mask for the legs
89
  parse_mask_legs = (parse_array == 12).astype(np.float32) + \
90
  (parse_array == 13).astype(np.float32) + \
91
+ (parse_array == 6).astype(np.float32) +\
92
+ (parse_array == 5).astype(np.float32)
 
93
 
94
  # Fill gaps between legs
95
  leg_mask_filled = cv2.dilate(parse_mask_legs.astype(np.uint8), np.ones((6, 6), np.uint8), iterations=6)