Pad zeros to a compressed keypoints array. Args: compressed_array (np.ndarray): A compressed keypoints array. mask_array (np.ndarray): The mask records compression relationship. Returns: np.ndarray:
(cls, compressed_array: np.ndarray,
mask_array: np.ndarray)
| 1299 | |
| 1300 | @classmethod |
| 1301 | def __add_zero_pad__(cls, compressed_array: np.ndarray, |
| 1302 | mask_array: np.ndarray) -> np.ndarray: |
| 1303 | """Pad zeros to a compressed keypoints array. |
| 1304 | |
| 1305 | Args: |
| 1306 | compressed_array (np.ndarray): |
| 1307 | A compressed keypoints array. |
| 1308 | mask_array (np.ndarray): |
| 1309 | The mask records compression relationship. |
| 1310 | |
| 1311 | Returns: |
| 1312 | np.ndarray: |
| 1313 | A keypoints array in full-size. |
| 1314 | """ |
| 1315 | assert mask_array.sum() == compressed_array.shape[1] |
| 1316 | data_len, _, dim = compressed_array.shape |
| 1317 | mask_len = mask_array.shape[0] |
| 1318 | ret_value = np.zeros(shape=[data_len, mask_len, dim], |
| 1319 | dtype=compressed_array.dtype) |
| 1320 | valid_mask_index = np.where(mask_array == 1)[0] |
| 1321 | ret_value[:, valid_mask_index, :] = compressed_array |
| 1322 | return ret_value |
| 1323 | |
| 1324 | @classmethod |
| 1325 | def __remove_zero_pad__(cls, zero_pad_array: np.ndarray, |
no outgoing calls
no test coverage detected