Init

Author: ZFTurbo

README.md

@@ -17,7 +17,7 @@ Coordinates for boxes expected to be normalized e.g in range [0; 1]. Order: x1,
 #### NMS, Soft-NMS, NMW
 
 ```
-from ensemble_boxes_nms import *
+from ensemble_boxes import *
 # Ensemble for several models predictions
 boxes, scores, labels = nms_method(boxes_list, scores_list, labels_list, method=method, weights=weights, iou_thr=iou_thr, sigma=sigma, thresh=thresh)
 # Ensemble for single model predictions
@@ -27,7 +27,7 @@ boxes, scores, labels = nms_method([boxes_list], [scores_list], [labels_list], w
 #### WBF
 
 ```
-from ensemble_boxes_wbf import *
+from ensemble_boxes import *
 # Ensemble for several models predictions
 boxes, scores, labels = weighted_boxes_fusion(boxes_list, scores_list, labels_list, method=method, weights=weights, iou_thr=iou_thr, sigma=sigma, thresh=thresh)
 # Ensemble for single model predictions
@@ -65,4 +65,4 @@ Ensemble script for them is available here: [example_oid.py](./example_oid.py)
 
 ## Description of WBF method
 
-_Details later_
+_Details later (arxiv link)_

example_oid.py

@@ -295,11 +295,11 @@ def ensemble_predictions(pred_filenames, weights, params):
 
     annotations_path = 'test_data/challenge-2019-validation-detection-bbox-expand_3520.csv'
     pred_list = [
-        'test_data/0.46450_TF_IRV2_atrous_nms_iou_0.3_valid_3520.csv',
-        'test_data/0.52319_mmdet_ensemble_wbf_all_levels_models_2_thr_1e-08_iou_0.55_1_1_valid_3520.csv',
-        'test_data/0.52918_ensemble_wbf_v2_all_levels_csvbased_2_thr_1e-09_iou_0.55_type_avg_valid_3520.csv',
-        'test_data/0.53775_ensemble_wbf_v2_all_levels_csvbased_2_thr_1e-09_iou_0.55_type_avg_valid_3520.csv',
-        'test_data/0.51145_zfturbo_ensemble_by_labels_5_valid_0.01_3520.csv',
+        'test_data/0.46450_TF_IRV2_atrous_3520.csv',
+        'test_data/0.52319_mmdet_3520.csv',
+        'test_data/0.52918_tensorpack1_3520.csv',
+        'test_data/0.53775_tensorpack2_3520.csv',
+        'test_data/0.51145_retinanet_3520.csv',
     ]
     weights = [1, 1, 1, 1, 1]