def get_dense_matrix(self, data: torch.tensor, c: torch.tensor):
batch_size = c[-1, -1] + 1
data = spconv.SparseConvTensor(data.unsqueeze(1),
c[:, self.permute_tensor],
self.spatial_size, batch_size)
data = data.dense()
data = data.detach().cpu().numpy()
return data
def get_dense_matrix(self,
data: torch.tensor,
c: torch.tensor,
to_numpy=True):
batch_size = c[-1, -1] + 1
if data.dim() == 1:
data = spconv.SparseConvTensor(data.unsqueeze(1),
c[:, self.permute_tensor],
self.spatial_size, batch_size)
else:
data = spconv.SparseConvTensor(data, c[:, self.permute_tensor],
self.spatial_size, batch_size)
data = data.dense()
if to_numpy:
data = data.detach().cpu().numpy()
return data
