_C.FDDB_DIR = '/home/lj/data/FDDB' _C.WIDER_DIR = '/home/lj/data/WIDER' _C.AFW_DIR = '/home/lj/data/AFW' _C.PASCAL_DIR = '/home/lj/data/PASCAL_FACE' # train config _C.MAX_STEPS = 120000 _C.LR_STEPS = (80000,100000,120000) _C.EPOCHES = 300 # anchor config _C.FEATURE_MAPS = [32, 16, 8] _C.STEPS = [32,64,128] _C.DENSITY = [[-3, -1, 1, 3], [-2, 2], [0]] _C.ASPECT_RATIOS = ((1, 2, 4), (1,), (1,)) _C.ANCHOR_SIZES = [32, 256, 512] _C.VARIANCE = [0.1, 0.2] _C.CLIP = False # loss config _C.NUM_CLASSES = 2 _C.OVERLAP_THRESH = 0.35 _C.NEG_POS_RATION = 7 # detection config _C.NMS_THRESH = 0.3 _C.NMS_TOP_K = 5000 _C.KEEP_TOP_K = 750 _C.CONF_THRESH = 0.05
_C.LABEL_MAP = dict( zip([str(x) for x in range(len(_C.VOC_CLASSES))], _C.VOC_CLASSES)) # train config _C.EPOCHES = 300 _C.LR_STEPS = (80000, 100000, 120000) _C.MAX_STEPS = 150000 # anchor config _C.SIZE320 = EasyDict() _C.FEATURE_MAPS = [40, 20, 10, 5] _C.INPUT_SIZE = 320 _C.STEPS = [8, 16, 32, 64] _C.MIN_SIZES = [32, 64, 128, 256] _C.MAX_SIZES = [64, 128, 256, 315] _C.ASPECT_RATIOS = [[2], [2], [2], [2]] _C.VARIANCE = [0.1, 0.2] _C.CLIP = True _C.NAME = 'VOC' # loss config _C.NUM_CLASSES = 21 _C.OVERLAP_THRESH = 0.5 _C.NEG_POS_RATIOS = 3 ## detection config _C.NMS_THRESH = 0.45 _C.NMS_TOP_K = 1000 _C.KEEP_TOP_K = 500 _C.CONF_THRESH = 0.01
import math from easydict import EasyDict __C = EasyDict() cfg = __C __C.DATASET_NAME = '' __C.INPUT_SIZE = (600, 600) __C.CLASS_NUM = 20 __C.CLASSES = ('aeroplane', 'bicycle', 'bird', 'boat', 'bottle', 'bus', 'car', 'cat', 'chair', 'cow', 'diningtable', 'dog', 'horse', 'motorbike', 'person', 'pottedplant', 'sheep', 'sofa', 'train', 'tvmonitor') __C.ANCHOR_SIZES = [32, 64, 128, 256, 512] __C.ASPECT_RATIOS = [0.5, 1.0, 2.0] __C.SCALE_RATIOS = [pow(2, 0 / 3.), pow(2, 1 / 3.), pow(2, 2 / 3.)]