From 47591cc637f285a253facc674c36d9e37629ec13 Mon Sep 17 00:00:00 2001 From: HuguesTHOMAS Date: Tue, 2 Mar 2021 09:53:24 -0500 Subject: [PATCH] Correcting num classes for cloud segmentation datasets --- datasets/S3DIS.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/datasets/S3DIS.py b/datasets/S3DIS.py index dfbcf51..227d4ac 100644 --- 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