Safran/KPConv-PyTorch
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

correcting bug in SemanticKitti

Browse source
This commit is contained in:
HuguesTHOMAS 2021-07-29 18:09:36 +00:00
parent 99bca4d438
commit 7fdbc57f9b
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
datasets/SemanticKitti.py
View file

@ -770,7 +770,7 @@ class SemanticKittiSampler(Sampler):
else: else:
class_indices = torch.zeros((0,), dtype=torch.int32) class_indices = torch.zeros((0,), dtype=torch.int32)
while class_indices.shape[0] < class_n: while class_indices.shape[0] < class_n:
new_class_inds = torch.randperm(class_potentials.shape[0]) new_class_inds = torch.randperm(class_potentials.shape[0]).type(torch.int32)
class_indices = torch.cat((class_indices, new_class_inds), dim=0) class_indices = torch.cat((class_indices, new_class_inds), dim=0)
class_indices = class_indices[:class_n] class_indices = class_indices[:class_n]
class_indices = self.dataset.class_frames[i][class_indices] class_indices = self.dataset.class_frames[i][class_indices]