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:
(self, compressed_array: np.ndarray,
mask_array: np.ndarray)
| 1272 | return decompressed_dict |
| 1273 | |
| 1274 | def add_zero_pad(self, compressed_array: np.ndarray, |
| 1275 | mask_array: np.ndarray) -> np.ndarray: |
| 1276 | """Pad zeros to a compressed keypoints array. |
| 1277 | |
| 1278 | Args: |
| 1279 | compressed_array (np.ndarray): |
| 1280 | A compressed keypoints array. |
| 1281 | mask_array (np.ndarray): |
| 1282 | The mask records compression relationship. |
| 1283 | |
| 1284 | Returns: |
| 1285 | np.ndarray: |
| 1286 | A keypoints array in full-size. |
| 1287 | """ |
| 1288 | assert mask_array.sum() == compressed_array.shape[1] |
| 1289 | data_len, _, dim = compressed_array.shape |
| 1290 | mask_len = mask_array.shape[0] |
| 1291 | ret_value = np.zeros(shape=[data_len, mask_len, dim], |
| 1292 | dtype=compressed_array.dtype) |
| 1293 | valid_mask_index = np.where(mask_array == 1)[0] |
| 1294 | ret_value[:, valid_mask_index, :] = compressed_array |
| 1295 | return ret_value |
no outgoing calls
no test coverage detected