patrickligardes commited on
Commit
8ef9858
ยท
verified ยท
1 Parent(s): aba8f51

Update utils_mask.py

Browse files
Files changed (1) hide show
  1. utils_mask.py +1 -1
utils_mask.py CHANGED
@@ -86,7 +86,7 @@ def get_mask_location(model_type, category, model_parse: Image.Image, keypoint:
86
  parser_mask_changeable += np.logical_and(parse_array, np.logical_not(parser_mask_fixed))
87
 
88
  # Include leg parts for gap filling
89
- parse_mask += (parse_array == 12).astype(np.float32) + \
90
  (parse_array == 13).astype(np.float32)
91
 
92
  # Fill gaps between legs
 
86
  parser_mask_changeable += np.logical_and(parse_array, np.logical_not(parser_mask_fixed))
87
 
88
  # Include leg parts for gap filling
89
+ parse_mask = (parse_array == 12).astype(np.float32) + \
90
  (parse_array == 13).astype(np.float32)
91
 
92
  # Fill gaps between legs