from easydict import EasyDict as EDict conf = EDict() conf.CLASS_NUM = 21 # in this case voc dataset. conf.MEAN_RGB = (123, 117, 104) #RGB not BGR conf.WD = 5e-4 conf.MOMENTUM = 0.9 conf.WORKSPACE = 512 conf.DOWN_SAMPLE_SCALE = 8 #train init model conf.LR_INIT = 16e-4 conf.EPOCH_SIZE_INIT = 200 conf.MAX_EPOCH_INIT = 40 conf.BATCH_SIZE_INIT = 16 conf.CROP_SIZE_INIT = 320 conf.SCALE_RANGE_INIT = [0.7, 1.3] #train final model conf.LR_FINAL = 16e-4 conf.EPOCH_SIZE_FINAL = 700 conf.MAX_EPOCH_FINAL = 40 conf.BATCH_SIZE_FINAL = 16 conf.CROP_SIZE_FINAL = 320 conf.SCALE_RANGE_FINAL = [0.7, 1.3] #for evaluate init and final models conf.CPU_WORKER_NUM = 8 conf.EVAL_WAIT_TIME = 0.3 # hour
