# 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