SPIGA/spiga/demo/analyze/track/retinasort/config.py

cfg_retinasort = {

    'retina': {
        'model_name': 'mobile0.25',
        'extra_features': ['landmarks'],
        'postreat': {
            'resize': 1.,
            'score_thr': 0.75,
            'top_k': 5000,
            'nms_thr': 0.4,
            'keep_top_k': 50}
        },

    'sort': {
        'max_age': 1,
        'min_hits': 3,
        'iou_threshold': 0.3,
    }
}

cfg_retinasort_res50 = {

    'retina': {
        'model_name': 'resnet50',
        'extra_features': ['landmarks'],
        'postreat': {
            'resize': 1.,
            'score_thr': 0.75,
            'top_k': 5000,
            'nms_thr': 0.4,
            'keep_top_k': 50}
        },

    'sort': {
        'max_age': 1,
        'min_hits': 3,
        'iou_threshold': 0.3,
    }
}

cfg_retinasort_cav3d = {

    'retina': {
        'model_name': 'resnet50',
        'extra_features': ['landmarks'],
        'postreat': {
            'resize': 1.,
            'score_thr': 0.95,
            'top_k': 5000,
            'nms_thr': 0.8,
            'keep_top_k': 50}
        },

    'sort': {
        'max_age': 90,
        'min_hits': 3,
        'iou_threshold': 0.3,
    }
}

cfg_retinasort_av16 = {

    'retina': {
        'model_name': 'resnet50',
        'extra_features': ['landmarks'],
        'postreat': {
            'resize': 1.,
            'score_thr': 0.75,
            'top_k': 5000,
            'nms_thr': 0.8,
            'keep_top_k': 50}
        },

    'sort': {
        'max_age': 90,
        'min_hits': 3,
        'iou_threshold': 0.3,
    }
}