Spaces:
Runtime error
Runtime error
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:
|