conf.OUTPUT_FOLDER = "outputs" conf.CLASS_NUM = 21 # in this case voc dataset. conf.MEAN_RGB = (123, 117, 104) #RGB not BGR #training params for FG BG cue networks. conf.EPOCH_SIZE_FG = 8000 conf.BATCH_SIZE_FG = 15 conf.LR_FG = 1e-3 conf.LR_DECAY_FG = 2000 conf.FG_CUE_FILE = "fg_cue_initsec.p" conf.SALIENCY_TH_FG = 0.2 conf.LR_BG = 1e-3 conf.EPOCH_SIZE_BG = 8000 conf.BATCH_SIZE_BG = 15 conf.LD_DECAY_BG = 2000 conf.BG_CUE_FILE = "bg_cue_initsec.p" conf.SALIENCY_TH_BG = 0.1 #training params for SEC models conf.CUE_FILE_INITSEC = "sec_cue.p" conf.INPUT_SIZE_SEC = 320 conf.DOWN_SAMPLE_SEC = 8 #network resolution conf.Q_FG = 0.996 conf.Q_BG = 0.999 conf.LR = 1e-3 conf.LR_DECAY = 2000 conf.MAX_EPOCH = 8 conf.EPOCH_SIZE = 1000
