Grid Clusters.

utils/project_config.py

@@ -11,6 +11,9 @@ class DataClass(Namespace):
     def __dict__(self):
         return {key: val for key, val in self.__class__.__dict__.items() if '__' not in key}
 
+    def items(self):
+        return self.__dict__().items()
+
     def __repr__(self):
         return f'{self.__class__.__name__}({self.__dict__().__repr__()})'
 
@@ -28,8 +31,8 @@ class Classes(DataClass):
 
 class DataSplit(DataClass):
     # DATA SPLIT OPTIONS
-    train = 'train',
-    devel = 'devel',
+    train = 'train'
+    devel = 'devel'
     test = 'test'
 
 
@@ -42,6 +45,10 @@ class GlobalVar(DataClass):
 
     classes = Classes()
 
+    grid_count = 12
+
+    prim_count = -1
+
 
 from models import *
 
@@ -50,4 +57,4 @@ class ThisConfig(Config):
 
     @property
     def _model_map(self):
-        return dict(PN2=PointNet2)
+        return dict(PN2=PointNet2, P2P=PointNet2PrimClusters, P2G=PointNet2GridClusters)