Пример #1
0
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
conf.MAX_INPUT_DIM = 800
conf.MULT_SCALE_EVAL = [0.7, 1.0, 1.3]  #multi scale evaluation can be used

conf.CRF_POS_XY_STD = 2
conf.CRF_POS_W = 3
conf.CRF_BI_RGB_STD = 3
conf.CRF_BI_XY_STD = 55
conf.CRF_BI_W = 4