configure WBF to mmdetection

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.74k stars 238 forks source link

Open Lensty opened 3 years ago

Lensty commented 3 years ago

How do I configure WBF to mmdetection

ZFTurbo commented 3 years ago

You can make inference on mmdetection models and store boxes as lists in memory. Then apply WBF function.

WBF is only effective for ensembles or TTA. So it's not replacement for NMS or Soft-NMS in single model.

ozanpkr commented 3 years ago

How do I configure WBF to mmdetection

Did you try this method with mmdetection json results?