CONFIGS.EXTRANET = CfgNode() CONFIGS.EXTRANET.NUMLAYERS = 2 CONFIGS.EXTRANET.NUMCONVS = 1 CONFIGS.EXTRANET.USE_INV_RES = False CONFIGS.EXTRANET.EXPAND_RATIO = 1 CONFIGS.DEEPBOX3D = CfgNode() CONFIGS.DEEPBOX3D.BINS = 2 CONFIGS.DEEPBOX3D.OVERLAP = 0.1 CONFIGS.DEEPBOX3D.OUT_CHANNELS = 256 CONFIGS.DEEPBOX3D.ALPHA = 0.6 CONFIGS.DEEPBOX3D.W = 0.4 CONFIGS.CAMERA = CfgNode() CONFIGS.CAMERA.KMAT = [ 7.188560e+02, 0.000000e+00, 6.071928e+02, 4.538225e+01, 0.000000e+00, 7.188560e+02, 1.852157e+02, -1.130887e-01, 0.000000e+00, 0.000000e+00, 1.000000e+00, 3.779761e-03 ] CONFIGS.DETECTOR = CfgNode() CONFIGS.DETECTOR.CHECKPOINT = './logdirs/mobiv2_deepbox3d/deepbox3d_mobilev2_980000.pkl' CONFIGS.INPUT = CfgNode() CONFIGS.INPUT.FORMAT = 'BGR' @Register.Config.register('MOBI-V2-DEEPBOX3D') def register(): return CONFIGS
CONFIGS.SOLVER.GAMMA = 0.1 # The iteration number to decrease learning rate by GAMMA. CONFIGS.SOLVER.STEPS = (20000, 50000) CONFIGS.SOLVER.WARMUP_FACTOR = 1.0 / 1000 CONFIGS.SOLVER.WARMUP_ITERS = 1000 CONFIGS.SOLVER.WARMUP_METHOD = "linear" CONFIGS.SOLVER.EXCLUDE_SCOPE = () CONFIGS.SOLVER.INCLUDE_SCOPE = () CONFIGS.SOLVER.LOAD_SOLVER = True CONFIGS.SOLVER.BIAS_LR_FACTOR = 1.0 CONFIGS.SOLVER.WEIGHT_DECAY_BIAS = CONFIGS.SOLVER.WEIGHT_DECAY CONFIGS.MODEL = CfgNode(new_allowed=True) CONFIGS.MODEL.BACKBONE = 'DLA-34' CONFIGS.MODEL.DOWN_SAMPLE = 4. CONFIGS.MODEL.OUT_CHANNELS = 256 CONFIGS.MODEL.KFNs = ['level2', 'level3', 'level4', 'level5'] CONFIGS.MODEL.FOCAL_LOSS_ALPHA = 2. CONFIGS.MODEL.FOCAL_LOSS_BEDA = 4. CONFIGS.MODEL.SMOOTH_L1_LOSS_BETA = 0.1 CONFIGS.MODEL.HEADER_NUM_CONV = 2 CONFIGS.DETECTOR = CfgNode(new_allowed=True) CONFIGS.DETECTOR.CHECKPOINT = './weights/DLA-34/model_0000004.pt' CONFIGS.DETECTOR.SCORE_THRESH = 0.5 CONFIGS.DETECTOR.TOPK_CANDIDATES = 30 CONFIGS.DETECTOR.NMS_THRESH_TEST = 0.5