steffen/Labeling-Primitives-With-Point2Net
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

redo

Browse Source
This commit is contained in:
Si11ium
2019-08-05 17:53:50 +02:00
parent 30525c954e
commit 54a5b48ddc
4 changed files with 31 additions and 161 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
main.py
View File

@ -73,12 +73,12 @@ if __name__ == '__main__':
test_transform = GT.Compose([GT.NormalizeScale(), ])
dataset = ShapeNetPartSegDataset(root_dir=opt.dataset, collate_per_segment=opt.collate_per_segment,
train=True, transform=train_transform, npoints=opt.npoints,
mode='train', transform=train_transform, npoints=opt.npoints,
has_variations=opt.has_variations, headers=opt.headers)
dataLoader = DataLoader(dataset, batch_size=opt.batch_size, shuffle=True, num_workers=opt.num_workers)
test_dataset = ShapeNetPartSegDataset(root_dir=opt.dataset, collate_per_segment=opt.collate_per_segment,
train=False, transform=test_transform, npoints=opt.npoints,
mode='test', transform=test_transform, npoints=opt.npoints,
has_variations=opt.has_variations, headers=opt.headers)
test_dataLoader = DataLoader(test_dataset, batch_size=opt.batch_size, shuffle=True, num_workers=opt.num_workers)