Weighted boxes fusion: Ensembling boxes from different object detection models

Roman Solovyev,Weimin Wang,Tatiana Gabruseva
DOI: https://doi.org/10.1016/j.imavis.2021.104117
2021-02-06
Abstract:In this work, we present a novel method for combining predictions of object detection models: weighted boxes fusion. Our algorithm utilizes confidence scores of all proposed bounding boxes to constructs the averaged boxes. We tested method on several datasets and evaluated it in the context of the Open Images and COCO Object Detection tracks, achieving top results in these challenges. The source code is publicly available at <a class="link-external link-https" href="https://github.com/ZFTurbo/Weighted-Boxes-Fusion" rel="external noopener nofollow">this https URL</a>
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?