cfg.FEAT_STRIDE = [8, 16]
cfg.DET_SCALES = [[8, 16], [16, 20]] # size: feat_stride * scale
cfg.DET_RATIOS = [1, 2, 3] # height / width
cfg.DET_CATE = {
'blouse': 0,
'skirt': 1,
'outwear': 2,
'dress': 2,
'trousers': 1,
}
cfg.PIXEL_MEAN = [0.485, 0.456, 0.406] # RGB
cfg.PIXEL_STD = [0.229, 0.224, 0.225]
cfg.BACKBONE_v2 = {
'resnet50': (gl.model_zoo.vision.resnet50_v1, ['stage2_activation3'], []),
}
cfg.BACKBONE_v3 = {
'resnet50':
(gl.model_zoo.vision.resnet50_v1,
['stage1_activation2', 'stage2_activation3', 'stage3_activation5'], []),
'resnet101':
(gl.model_zoo.vision.resnet101_v1,
['stage1_activation2', 'stage2_activation3', 'stage3_activation22'], []),
}
cfg.BACKBONE_v4 = {
'resnet50': (gl.model_zoo.vision.resnet50_v1, [
'conv1', 'stage1_activation2', 'stage2_activation3',
'stage3_activation5'
