- 博客(2)
- 收藏
- 关注
原创 WBF模型融合
作者认为,在目标检测任务中,当实时性要求不强时,集成多个推理模型有助于我们得到更好的结果。于是作者在文中提出了一种方法,可以把多个目标检测模型的预测结果结合起来。这个方法称为 weighted boxes fusion,简称 WBF。在筛选预测框的过程中,常用的方法是非极大值抑制(non-maximum, NMS),还有一种 soft-NMS 方法也被用于改进筛选性能。这些方法在单个模型上效果不错,但只能简单地从预测框中,“筛选” 框框。不能有效地结合多个模型的预测结果来产生一个平均的框框。
2023-08-15 11:22:20
284
2
原创 SSD目标检测复现
tain.py代码中,在迭代过程中,每次执行batch张图片,通过images, targets = next(batch_iterator)读取图片时,如果next()中没有数据后会触发Stoplteration异常,使用下面语句替换 images, targets = next(batch_iterator)将解决这种异常问题。注释掉的是源代码,下面的是新加的,即交换两行代码的顺序即可。只用voc2007数据集,代码默认先使用coco数据集,将coco部分注释掉。解决了这个问题,又会出现新的问题。
2022-09-02 10:08:48
1989
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人