BACKBONE.STAGE3.BLOCK = 'BASIC'
BACKBONE.STAGE3.FUSE_METHOD = 'SUM'
BACKBONE.STAGE4 = EasyDict()
BACKBONE.STAGE4.NUM_MODULES = 3
BACKBONE.STAGE4.NUM_BRANCHES = 4
BACKBONE.STAGE4.NUM_BLOCKS = [4, 4, 4, 4]
BACKBONE.STAGE4.NUM_CHANNELS = [32, 64, 128, 256]
BACKBONE.STAGE4.BLOCK = 'BASIC'
BACKBONE.STAGE4.FUSE_METHOD = 'SUM'
NECK = EasyDict()
NECK.IN_CHANNEL = [32, 64, 128, 256]
HEAD = EasyDict()
HEAD.BACKBONE = 'hrnet'
HEAD.IN_CHANNEL = [32, 64, 128, 256]
HEAD.OUT_HEADS = {
'yolo-0':
len(DATASET.ANCHORS[0]) * (dataset[DATASET.NAME]['num_class'] + 5),
'yolo-1':
len(DATASET.ANCHORS[0]) * (dataset[DATASET.NAME]['num_class'] + 5),
'yolo-2':
len(DATASET.ANCHORS[0]) * (dataset[DATASET.NAME]['num_class'] + 5),
'yolo-3':
len(DATASET.ANCHORS[0]) * (dataset[DATASET.NAME]['num_class'] + 5)
}
OPTIM = EasyDict()
OPTIM.BURN_IN = 1000
OPTIM.EPOCH = 300
0.13322]],
[[0.05088, 0.03728], [0.10116, 0.08054], [0.10545,
0.26739]],
[[0.21067, 0.15731], [0.27648, 0.42582], [0.69417,
0.71236]]]
BACKBONE = EasyDict()
BACKBONE.INIT_WEIGHTS = True
BACKBONE.PRETRAINED = osp.join(osp.dirname(__file__), '..',
'weights/pretrained/darknet53_74.pth')
NECK = EasyDict()
NECK.IN_CHANNEL = [256, 512, 1024]
HEAD = EasyDict()
HEAD.BACKBONE = 'darknet'
HEAD.IN_CHANNEL = [128, 256, 512]
HEAD.OUT_HEADS = {
'yolo-0':
len(DATASET.ANCHORS[0]) * (dataset[DATASET.NAME]['num_class'] + 5),
'yolo-1':
len(DATASET.ANCHORS[0]) * (dataset[DATASET.NAME]['num_class'] + 5),
'yolo-2':
len(DATASET.ANCHORS[0]) * (dataset[DATASET.NAME]['num_class'] + 5)
}
OPTIM = EasyDict()
OPTIM.BURN_IN = 1000
OPTIM.EPOCH = 200
OPTIM.INIT_LR = 1e-5
OPTIM.END_LR = 1e-6
