Exemplo n.º 1
0
Config.loss_gamma = 1.

# dataset
Config.num_classes = 20
Config.batch_size = 16  # 128
Config.root = '../../dataset/voc/VOCdevkit/VOC2012'
Config.split = 'trainval'
Config.resize_size = [512, 512]
Config.num_workers = 4
Config.mean = [0.40789654, 0.44719302, 0.47026115]
Config.std = [0.28863828, 0.27408164, 0.27809835]

# train
Config.optimizer = 'AdamW'
Config.lr = 1e-3
Config.AMSGRAD = True

Config.max_iter = 45000
Config.lr_schedule = 'WarmupMultiStepLR'
Config.gamma = 0.1
Config.steps = (35000, 40000)
Config.warmup_iters = 1000

Config.apex = False

# other
Config.gpu = True
Config.eval = False
Config.resume = False
Config.score_th = 0.1
Config.down_stride = 4