search

ZFTurbo / Weighted-Boxes-Fusion

Set of methods to ensemble boxes from different object detection models, including implementation of "Weighted boxes fusion (WBF)" method.
MIT License
1.72k stars 238 forks source link

single model #17

Open SISTMrL opened 4 years ago

SISTMrL commented 4 years ago

the wbf can be used for only single model? If I only use the yolov5 to get the result, can i apply the wbf to refine the detection?

ZFTurbo commented 4 years ago

In our experiments WBF for single model gave worse results than NMS. WBF is good for ensemble of TTA or different models.