diff --git a/dataset/shapenet.py b/dataset/shapenet.py index f63a857..27b9e96 100644 --- a/dataset/shapenet.py +++ b/dataset/shapenet.py @@ -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: