CONFIGS.TRAINING.CHECKPOINT_MODE = 'PRETRAINED' #['PRETRAINED', 'RESUME', 'START']
CONFIGS.TRAINING.CHECKPOINT_FILE = './pretrained/mobilenetv2_pretrained.pkl'
CONFIGS.DATASET = CfgNode()
CONFIGS.DATASET.PATH = './datasets/data/kitti'
CONFIGS.DATASET.MEAN = [95.87739305, 98.76049672, 93.83309082]
CONFIGS.DATASET.DIM_MEAN = [[1.52607842, 1.62858147, 3.88396124],
[2.20649159, 1.90197734, 5.07812564],
[3.25207685, 2.58505032, 10.10703568],
[1.76067766, 0.6602296, 0.84220464],
[1.27538462, 0.59506787, 0.80180995],
[1.73712792, 0.59677122, 1.76338868],
[3.52905882, 2.54368627, 16.09707843],
[1.9074177, 1.51386831, 3.57683128]]
CONFIGS.DATALOADER = CfgNode()
CONFIGS.DATALOADER.SAMPLER_TRAIN = 'TrainingSampler'
# Solver
# ---------------------------------------------------------------------------- #
CONFIGS.SOLVER = CfgNode()
# See detectron2/solver/build.py for LR scheduler options
CONFIGS.SOLVER.LR_SCHEDULER_NAME = "WarmupMultiStepLR"
CONFIGS.SOLVER.MAX_ITER = 100000
CONFIGS.SOLVER.BASE_LR = 0.01
CONFIGS.SOLVER.MOMENTUM = 0.9
CONFIGS.SOLVER.WEIGHT_DECAY = 0.0001
_C.DATASETS.TRAIN = () _C.DATASETS.PROPOSAL_FILES_TRAIN = () _C.DATASETS.PRECOMPUTED_PROPOSAL_TOPK_TRAIN = 2000 _C.DATASETS.TEST = () _C.DATASETS.PROPOSAL_FILES_TEST = () _C.DATASETS.PRECOMPUTED_PROPOSAL_TOPK_TEST = 1000 # ----------------------------------------------------------------------------- # DataLoader options # ----------------------------------------------------------------------------- _C.DATALOADER = CN() _C.DATALOADER.NUM_WORKERS = 4 _C.DATALOADER.ASPECT_RATIO_GROUPING = True _C.DATALOADER.SAMPLER_TRAIN = "TrainingSampler" _C.DATALOADER.REPEAT_THRESHOLD = 0.0 _C.DATALOADER.FILTER_EMPTY_ANNOTATIONS = True # ----------------------------------------------------------------------------- # Solver options # ----------------------------------------------------------------------------- _C.SOLVER = CN()
