目标检测之 WBF（Weighted Boxes Fusion）



短短3页中了CVPR2019，可以算是那种为数不多的几行代码可以显著提升模型性能的论文了。简单有效值的一读。



论文地址：

https://arxiv.org/abs/1910.13302



代码地址：https://github.com/ZFTurbo/Weighted-Boxes-Fusion



具体步骤：





个人感觉论文里面讲的还是不够清晰。这里就这节上源码了。





这里首先对传入的参数做说明， 然后会通过prefilter_boxes()对所有的box 和score做处理构建一个label为key,[int(label), float(score) * weights[t], x1, y1, x2, y2] 为value的dict。



具体实现如下：





注意一下的是，这个dict的value最终是经过排序的。