34 lines
489 B
Python
34 lines
489 B
Python
from argparse import Namespace
|
|
|
|
from ml_lib.utils.config import Config
|
|
|
|
|
|
class GlobalVar(Namespace):
|
|
# Labels for classes
|
|
LEFT = 1
|
|
RIGHT = 0
|
|
WRONG = -1
|
|
|
|
# Colors for img files
|
|
WHITE = 255
|
|
BLACK = 0
|
|
|
|
# Variables for plotting
|
|
PADDING = 0.25
|
|
DPI = 50
|
|
|
|
# DATAOPTIONS
|
|
train ='train',
|
|
vali ='vali',
|
|
test ='test'
|
|
|
|
|
|
from models import *
|
|
|
|
|
|
class ThisConfig(Config):
|
|
|
|
@property
|
|
def _model_map(self):
|
|
return dict(PN2=PointNet2)
|