Пример #1
0
conf.VOC_TRAIN_IM_FOLDER = "train_images"
conf.VOC_VAL_IM_FOLDER = "val_images"
conf.VOC_VAL_MASK_FOLDER = "val_masks"
conf.VOC_TRAIN_LIST = "train_list.txt"
conf.VOC_VAL_LIST = "val_list.txt"

conf.LOG_FOLDER = "log"
conf.SNAPSHOT_FOLDER = "snapshots"
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