_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.)]
