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

.

Browse source
This commit is contained in:
HuguesTHOMAS 2021-07-29 16:35:35 +00:00
parent 1527e7d09e
commit 1fcd3bbf52
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
train_S3DIS.py
View file

@ -268,8 +268,8 @@ if __name__ == '__main__':
config.saving_path = sys.argv[1]
# Initialize datasets
training_dataset = S3DISDataset(config, set='training', use_potentials=False)
test_dataset = S3DISDataset(config, set='validation', use_potentials=False)
training_dataset = S3DISDataset(config, set='training', use_potentials=True)
test_dataset = S3DISDataset(config, set='validation', use_potentials=True)
# Initialize samplers
training_sampler = S3DISSampler(training_dataset)