Spaces:
Sleeping
Sleeping
full_body fixes
Browse files- 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 |
-
|
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:
|