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 #for dataset #SBD_PATH is the one named "dataset", which has "cls", "img", "inst", "train.txt" and "val.txt". #please download at https://www2.eecs.berkeley.edu/Research/Projects/CS/vision/grouping/semantic_contours/benchmark.tgz #VOCDEVKIT_PATH is the one named "VOC2012", which has "Annotations", "ImageSets", "JPEGImages", "SegmentationClass", conf.DATASET_PATH = "dataset" conf.SBD_PATH = "dataset/benchmark_RELEASE/dataset" conf.VOCDEVKIT_PATH = "dataset/VOCdevkit/VOC2012" conf.VOC_TRAIN_MULTI_FILE = "voc_multi_file.p" 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.FINAL_VOC_MASK_FOLDER = "final_voc_masks" conf.CACHE_PATH = "cache" conf.BASE_NET = "resnet50" conf.WEB_IMAGE_FOLDER = "web_images" conf.WEB_IMAGE_MASKS_FOLDER = "web_labels" conf.WEB_TRAIN_LIST = "web_train_list.txt" conf.LOG_FOLDER = "log" conf.SNAPSHOT_FOLDER = "snapshots" conf.OUTPUT_FOLDER = "outputs"
from easydict import EasyDict as EDict conf = EDict() conf.BASE_NET = "vgg16" #vgg16 or resnet50 conf.CACHE_PATH = "cache" #for dataset #SBD_PATH is the one named "dataset", which has "cls", "img", "inst", "train.txt" and "val.txt". #please download at https://www2.eecs.berkeley.edu/Research/Projects/CS/vision/grouping/semantic_contours/benchmark.tgz #VOCDEVKIT_PATH is the one named "VOC2012", which has "Annotations", "ImageSets", "JPEGImages", "SegmentationClass", conf.DATASET_PATH = "dataset" conf.SBD_PATH = "dataset/benchmark_RELEASE/dataset" conf.VOCDEVKIT_PATH = "dataset/VOCdevkit/VOC2012" conf.VOC_TRAIN_MULTI_FILE = "voc_multi_file.p" 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