Remove zero-padding from a full-size keypoints array. Args: zero_pad_array (np.ndarray): A keypoints array in full-size. mask_array (np.ndarray): The mask records compression relationship. Returns: np.ndarray:
(cls, zero_pad_array: np.ndarray,
mask_array: np.ndarray)
| 1323 | |
| 1324 | @classmethod |
| 1325 | def __remove_zero_pad__(cls, zero_pad_array: np.ndarray, |
| 1326 | mask_array: np.ndarray) -> np.ndarray: |
| 1327 | """Remove zero-padding from a full-size keypoints array. |
| 1328 | |
| 1329 | Args: |
| 1330 | zero_pad_array (np.ndarray): |
| 1331 | A keypoints array in full-size. |
| 1332 | mask_array (np.ndarray): |
| 1333 | The mask records compression relationship. |
| 1334 | |
| 1335 | Returns: |
| 1336 | np.ndarray: |
| 1337 | A compressed keypoints array. |
| 1338 | """ |
| 1339 | assert mask_array.shape[0] == zero_pad_array.shape[1] |
| 1340 | valid_mask_index = np.where(mask_array == 1)[0] |
| 1341 | ret_value = np.take(zero_pad_array, valid_mask_index, axis=1) |
| 1342 | return ret_value |
| 1343 | |
| 1344 | @classmethod |
| 1345 | def __get_key_warn_msg__(cls, key: Any) -> str: |
no outgoing calls
no test coverage detected