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

exclude some more*2

Browse Source
This commit is contained in:
Si11ium 2019-08-02 19:12:21 +02:00
parent 0a5be5d4da
commit 30525c954e
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dataset/shapenet.py
View File

@ -101,7 +101,7 @@ class CustomShapeNet(InMemoryDataset):
for element in paths: for element in paths:
# This was build to filter all variations that aregreater then 25 # This was build to filter all variations that aregreater then 25
pattern = re.compile('^((6[0-1]|[2-5][0-9])_\w+?\d+?|pc|\d+?_pc)\.(xyz|dat)$') pattern = re.compile('^((6[0-1]|[1-5][0-9])_\w+?\d+?|pc|\d+?_pc)\.(xyz|dat)$')
if pattern.match(os.path.split(element)[-1]): if pattern.match(os.path.split(element)[-1]):
continue continue
else: else:

3
main.py
View File

@ -123,7 +123,8 @@ if __name__ == '__main__':
print('Epoch {}, total epoches {}'.format(epoch+1, opt.nepoch)) print('Epoch {}, total epoches {}'.format(epoch+1, opt.nepoch))
net.train() net.train()
# ToDo: We need different dataloader here to train the network in multiple iterations, maybe move the loop down
# for dataloader in ...
for batch_idx, sample in enumerate(dataLoader): for batch_idx, sample in enumerate(dataLoader):
# points: (batch_size, n, 3) # points: (batch_size, n, 3)
# labels: (batch_size, n) # labels: (batch_size, n)