# detect config file
config = dict(

    data=dict(
        image_files='test.txt',  # 需要检测的图像 .txt/dir/.jpg/.png
    ),
    model=dict(
        cuda='cpu',
        class_txt='config/id/id_classes.txt',
        config_files=['config/id/cascade_rcnn_r101_fpn_1x_coco_multiscale.py', 'config/id/faster_rcnn_r101_fpn_2x_coco_multiscale.py'],
        checkpoint_files=['model/id/cascade_rcnn_r101_fpn_1x_multiscale_epoch_20.pth', 'model/id/faster_rcnn_r101_fpn_2x_multiscale_epoch_24.pth'],
    ),
    fusion=dict(
        type=['weighted_boxes_fusion'],
        class_list=[[1, 5], [0, 2, 3, 4]],
        weight_list=[1, 1],  # 不同模型所占权重
        iou=0.5,  # 融合iou设置
        score=0.4,
        skip_box_thr=0.0001  # 排除低于此阈值score的box
    ),
    show=False,
    save_txt=True,
    save_image=True,
    out_dir='../out/test_20220516'
)


def load_config():
    return config