From ec5baecd82a4856ada3db306d4276eca2c854512 Mon Sep 17 00:00:00 2001 From: ZFTurbo Date: Sun, 27 Oct 2019 19:47:41 +0300 Subject: [PATCH] Init --- README.md | 6 +++--- example_oid.py | 10 +++++----- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/README.md b/README.md index bc5c139..3ee910b 100644 --- a/README.md +++ b/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)_ diff --git a/example_oid.py b/example_oid.py index 3396607..60925de 100644 --- a/example_oid.py +++ b/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]