Correcting num classes for cloud segmentation datasets

2021-03-02 09:53:24 -05:00 · 2021-03-02 09:53:24 -05:00 · 47591cc637
parent 94b1d87e17
commit 47591cc637
1 changed files with 1 additions and 1 deletions
--- a/datasets/S3DIS.py
+++ b/datasets/S3DIS.py
@ -91,7 +91,7 @@ class S3DISDataset(PointCloudDataset):
        self.dataset_task = 'cloud_segmentation'

        # Update number of class and data task in configuration
-        config.num_classes = self.num_classes
+        config.num_classes = self.num_classes - len(self.ignored_labels)
        config.dataset_task = self.dataset_task

        # Parameters from config