Example #1
0
    (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'], []),
}

cfg.EVAL_NORMAL_IDX = {
    'blouse': (5, 6),
    'skirt': (15, 16),
    'outwear': (5, 6),
    'dress': (5, 6),
    'trousers': (15, 16),
}

cfg.SCALE_MIN = cfg.CROP_SIZE * cfg.SCALE_MIN_RATE
cfg.SCALE_MAX = cfg.CROP_SIZE * cfg.SCALE_MAX_RATE
cfg.PIXEL_MEAN = np.array(cfg.PIXEL_MEAN, dtype='float32').reshape((3, 1, 1))
cfg.PIXEL_STD = np.array(cfg.PIXEL_STD, dtype='float32').reshape((3, 1, 1))
cfg.FILL_VALUE = [int(v * 255)