Inference written

This commit is contained in:
Si11ium
2020-06-07 17:30:47 +02:00
parent 2a767bead2
commit 27ae8467fc
2 changed files with 84 additions and 0 deletions

View File

@ -17,6 +17,9 @@ class DataClass(Namespace):
def __repr__(self):
return f'{self.__class__.__name__}({self.__dict__().__repr__()})'
def __getitem__(self, item):
return self.__getattribute__(item)
class Classes(DataClass):
# Object Classes for Point Segmentation
@ -29,6 +32,12 @@ class Classes(DataClass):
Plane = 6
class Settings(DataClass):
P2G = 'grid'
P2P = 'prim'
PN2 = 'pc'
class DataSplit(DataClass):
# DATA SPLIT OPTIONS
train = 'train'
@ -49,6 +58,8 @@ class GlobalVar(DataClass):
prim_count = -1
settings = Settings()
from models import *