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

Correcting num classes for cloud segmentation datasets

Browse source
This commit is contained in:
HuguesTHOMAS 2021-03-02 09:53:24 -05:00
parent 94b1d87e17
commit 47591cc637
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/S3DIS.py
View file

@ -91,7 +91,7 @@ class S3DISDataset(PointCloudDataset):
self.dataset_task = 'cloud_segmentation' self.dataset_task = 'cloud_segmentation'
# Update number of class and data task in configuration # 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 config.dataset_task = self.dataset_task
# Parameters from config # Parameters from config