From 0a5be5d4da5ee60404a003584d6890adabd6bce7 Mon Sep 17 00:00:00 2001
From: Si11ium <steffen.illium@ifi.lmu.de>
Date: Fri, 2 Aug 2019 18:48:09 +0200
Subject: [PATCH] exclude some more
---
dataset/shapenet.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
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: