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

Now with included sorting per cloud variation - Some korrektions

Browse Source
This commit is contained in:
Si11ium 2019-08-02 13:13:43 +02:00
parent 43b4c8031a
commit 45d64e5b82
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
dataset/shapenet.py
View File

@ -24,6 +24,7 @@ class CustomShapeNet(InMemoryDataset):
def __init__(self, root, collate_per_segment=True, train=True, transform=None, pre_filter=None, pre_transform=None, def __init__(self, root, collate_per_segment=True, train=True, transform=None, pre_filter=None, pre_transform=None,
headers=True, has_variations=False): headers=True, has_variations=False):
assert not (collate_per_segment and has_variations), 'Either use each element or pointclouds - with variations'
self.has_headers = headers self.has_headers = headers
self.has_variations = has_variations self.has_variations = has_variations
self.collate_per_element = collate_per_segment self.collate_per_element = collate_per_segment