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

SemanticKitti inf while loop correction

Browse source
This commit is contained in:
HuguesTHOMAS 2021-07-29 21:26:45 +00:00
parent 7fdbc57f9b
commit 7f5f52b067
1 changed files with 21 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
datasets/SemanticKitti.py
View file

@ -762,6 +762,8 @@ class SemanticKittiSampler(Sampler):
# Get the potentials of the frames containing this class # Get the potentials of the frames containing this class
class_potentials = self.dataset.potentials[self.dataset.class_frames[i]] class_potentials = self.dataset.potentials[self.dataset.class_frames[i]]
if class_potentials.shape[0] > 0:
# Get the indices to generate thanks to potentials # Get the indices to generate thanks to potentials
used_classes = self.dataset.num_classes - len(self.dataset.ignored_labels) used_classes = self.dataset.num_classes - len(self.dataset.ignored_labels)
class_n = num_centers // used_classes + 1 class_n = num_centers // used_classes + 1