patrickligardes commited on
Commit
ef2edb7
ยท
verified ยท
1 Parent(s): f2a3222

Update utils_mask.py

Browse files
Files changed (1) hide show
  1. utils_mask.py +1 -1
utils_mask.py CHANGED
@@ -91,7 +91,7 @@ def get_mask_location(model_type, category, model_parse: Image.Image, keypoint:
91
  (parse_array == label_map["pants"]).astype(np.float32)
92
  parser_mask_fixed += parser_mask_fixed_lower_cloth
93
  parser_mask_changeable += np.logical_and(parse_array, np.logical_not(parser_mask_fixed))
94
- elif category == 'lower_body':
95
  parse_mask = (parse_array == 6).astype(np.float32) + \
96
  (parse_array == 12).astype(np.float32) + \
97
  (parse_array == 13).astype(np.float32) + \
 
91
  (parse_array == label_map["pants"]).astype(np.float32)
92
  parser_mask_fixed += parser_mask_fixed_lower_cloth
93
  parser_mask_changeable += np.logical_and(parse_array, np.logical_not(parser_mask_fixed))
94
+ elif category == 'lower_body' or category == 'full_body':
95
  parse_mask = (parse_array == 6).astype(np.float32) + \
96
  (parse_array == 12).astype(np.float32) + \
97
  (parse_array == 13).astype(np.float32) + \