spdraptor commited on
Commit
09e75d4
Β·
1 Parent(s): c3f7e1d

full_body fixes

Browse files
Files changed (1) hide show
  1. utils_mask.py +1 -5
utils_mask.py CHANGED
@@ -107,11 +107,7 @@ def get_mask_location(model_type, category, model_parse: Image.Image, keypoint:
107
  (parse_array == 12).astype(np.float32) + \
108
  (parse_array == 13).astype(np.float32) + \
109
  (parse_array == 5).astype(np.float32)
110
- parser_mask_fixed += (parse_array == label_map["skirt"]).astype(np.float32) + \
111
- (parse_array == label_map["pants"]).astype(np.float32) + \
112
- (parse_array == label_map["upper_clothes"]).astype(np.float32) + \
113
- (parse_array == 14).astype(np.float32) + \
114
- (parse_array == 15).astype(np.float32)
115
  parser_mask_changeable += np.logical_and(parse_array, np.logical_not(parser_mask_fixed))
116
 
117
  else:
 
107
  (parse_array == 12).astype(np.float32) + \
108
  (parse_array == 13).astype(np.float32) + \
109
  (parse_array == 5).astype(np.float32)
110
+
 
 
 
 
111
  parser_mask_changeable += np.logical_and(parse_array, np.logical_not(parser_mask_fixed))
112
 
113
  else: