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

Adding classification test method

Browse source
This commit is contained in:
HuguesTHOMAS 2020-08-11 10:20:52 -04:00
parent bbe199bb60
commit 94b1d87e17
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
test_models.py
View file

@ -162,7 +162,7 @@ if __name__ == '__main__':
set = 'test'
# Initiate dataset
if config.dataset.startswith('ModelNet40'):
if config.dataset == 'ModelNet40':
test_dataset = ModelNet40Dataset(config, train=False)
test_sampler = ModelNet40Sampler(test_dataset)
collate_fn = ModelNet40Collate
@ -209,7 +209,7 @@ if __name__ == '__main__':
# Training
if config.dataset_task == 'classification':
a = 1/0
tester.classification_test(net, test_loader, config)
elif config.dataset_task == 'cloud_segmentation':
tester.cloud_segmentation_test(net, test_loader, config)
elif config.dataset_task == 'slam_segmentation':