GitHub - ZFTurbo/Weighted-Boxes-Fusion: Set of methods to ensemble boxes from different object detection models, including implementation of "Weighted boxes fusion (WBF)" method.

Coordinates for boxes expected to be normalized e.g in range [0; 1]. Order: x1, y1, x2, y2. Example of boxes ensembling for 2 models below. First model predicts 5 boxes, second model predicts 4 boxes. Confidence scores for each box model 1: [0.9, 0.8, 0.2, 0.4, 0.7] Confidence scores for each box model 2: [0.5, 0.8, 0.7, 0.3] Labels (classes) for each box model 1: [0, 1, 0, 1, 1] Labels (classes)