Skip to content

Commit ee82a3a

Browse files
committed
Added support for padded point clouds
1 parent c86d7f6 commit ee82a3a

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

dect/ect.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -178,6 +178,8 @@ def compute_ect_point_cloud(
178178
start=-radius, end=radius, steps=resolution, device=x.device
179179
).view(-1, 1, 1)
180180
nh = (x @ v).unsqueeze(1)
181+
nh[nh.isnan()] = torch.inf
182+
nh[nh.isinf()] = torch.inf
181183
ecc = torch.nn.functional.sigmoid(scale * torch.sub(lin, nh))
182184
ect = torch.sum(ecc, dim=2)
183185
if normalize:

0 commit comments

Comments
 (0)