Пример #1
0
    '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'
    ], []),
}

cfg.BACKBONE_Det = {
    'resnet50': (gl.model_zoo.vision.resnet50_v1,
                 ['stage2_activation3', 'stage3_activation5'], []),
}