dataset = {
'coco': {
'path': '/data/yoloCao/DataSet/person_helmet/coco2017',
'num_class': 80
},
'pascal': {
'path': '/data/yoloCao/DataSet/VOC',
'num_class': 20
},
'objects365': '',
}
DATASET = EasyDict()
DATASET.NAME = 'pascal'
DATASET.DATA_DIR = dataset[DATASET.NAME]['path']
DATASET.BATCHSIZE = 14
DATASET.DOWN_RATION = [4, 8, 16, 32]
DATASET.INPUT_H = 416
DATASET.INPUT_W = 416
DATASET.USE_GIOU = False
DATASET.EVAL = False
DATASET.ANCHORS = [[[0.01181, 0.01955], [0.01941, 0.05309], [0.03584, 0.1]],
[[0.03831, 0.02876], [0.06861, 0.05756], [0.06994,
0.16855]],
[[0.11966, 0.3311], [0.13606, 0.0961], [0.21739, 0.1955]],
[[0.25938, 0.5573], [0.47191, 0.32135], [0.72506, 0.77527]]]
BACKBONE = EasyDict()
BACKBONE.INIT_WEIGHTS = True
BACKBONE.PRETRAINED = osp.join(osp.dirname(__file__), '..',
'weights/pretrained/hrnet_w32-36af842e.pth')
