SemanticKitti inf while loop correction

2021-07-29 21:26:45 +00:00 · 2021-07-29 21:26:45 +00:00 · 7f5f52b067
parent 7fdbc57f9b
commit 7f5f52b067
1 changed files with 21 additions and 19 deletions
--- a/datasets/SemanticKitti.py
+++ b/datasets/SemanticKitti.py
@ -762,6 +762,8 @@ class SemanticKittiSampler(Sampler):
                    # Get the potentials of the frames containing this class
                    class_potentials = self.dataset.potentials[self.dataset.class_frames[i]]

+                    if class_potentials.shape[0] > 0:
+
                        # Get the indices to generate thanks to potentials
                        used_classes = self.dataset.num_classes - len(self.dataset.ignored_labels)
                        class_n = num_centers // used_classes + 1