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

exclude some more

Browse Source
This commit is contained in:
Si11ium 2019-08-02 18:48:09 +02:00
parent 376d8f0d7c
commit 0a5be5d4da
1 changed files with 1 additions and 1 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:
# This was build to filter all variations that aregreater then 25
pattern = re.compile('^((2[5-9]|[3-5][0-9])_\w+?\d+?|pc|\d+?_pc)\.(xyz|dat)$')
pattern = re.compile('^((6[0-1]|[2-5][0-9])_\w+?\d+?|pc|\d+?_pc)\.(xyz|dat)$')
if pattern.match(os.path.split(element)[-1]):
continue
else: