get bbox of kp2d.
(keypoint_np: np.ndarray,
img_mat: Optional[np.ndarray] = None,
expand: bool = False)
| 165 | |
| 166 | |
| 167 | def _get_bbox(keypoint_np: np.ndarray, |
| 168 | img_mat: Optional[np.ndarray] = None, |
| 169 | expand: bool = False): |
| 170 | """get bbox of kp2d.""" |
| 171 | x_max = np.max(keypoint_np[:, 0]) |
| 172 | x_min = np.min(keypoint_np[:, 0]) |
| 173 | y_max = np.max(keypoint_np[:, 1]) |
| 174 | y_min = np.min(keypoint_np[:, 1]) |
| 175 | if expand and img_mat is not None: |
| 176 | x_expand = (x_max - x_min) * 0.1 |
| 177 | y_expand = (y_max - y_min) * 0.1 |
| 178 | x_min = max(0, x_min - x_expand) |
| 179 | x_max = min(img_mat.shape[1], x_max + x_expand) |
| 180 | y_min = max(0, y_min - y_expand) |
| 181 | y_max = min(img_mat.shape[0], y_max + y_expand) |
| 182 | return np.asarray([x_min, x_max, y_min, y_max]) |
| 183 | |
| 184 | |
| 185 | def _prepare_limb_palette(limbs, |