首页
学习
活动
专区
圈层
工具
发布

新算法 | 基于DIou改进的YOLOv3目标检测

DIoU-NMS倾向于中心点距离较远的box存在不同的对象,而且仅需改几行代码,DIoU-NMS就能够很简单地集成到目标检测算法中。 ? 实验结果 ?...mAP分别提升0.59%和0.84%,而AP75则分别提升1.77%和2.59%,而使用DIoU-NMS则能进一步提升效果。 Faster R-CNN on MS COCO ? ?...Discussion on DIoU-NMS ? DIoU-NMS能够更好地引导bbox的消除 ?...最后作者进行了比较实验,原NMS和DIoU-NMS分别移植到了yolov3和ssd模型,同时使用CIou loss。在阈值的范围内比较了原始NMS和DIoU-NMS ? [0.43,0.48]。...从图9中可以看出,对于每个阈值,DIoU-NMS都优于原始NMS。此外,值得注意的是,即使是最糟糕的DIoU-NMS也是要优于最优的原始NMS。 ?

1.9K20
  • 您找到你想要的搜索结果了吗?
    是的
    没有找到

    NMS Soft-NMS DIoU-NMS:目标检测后处理技术演进

    本文深入剖析NMS的工作原理、局限性,以及Soft-NMS、DIoU-NMS等改进算法的创新点。通过对比分析和代码实现,展示不同NMS变体在密集场景下的性能差异,并探讨其在YOLO系列中的应用。...随着YOLO系列的不断演进,研究者们提出了多种NMS改进算法,如Soft-NMS、DIoU-NMS、CIoU-NMS等,旨在解决这些问题。...2.3 DIoU-NMS:考虑距离信息的抑制策略 DIoU-NMS是在DIoU损失函数基础上提出的改进算法,其核心思想是在计算抑制权重时,不仅考虑IoU,还考虑两个框的中心点距离。...具体来说,DIoU-NMS通过以下公式更新置信度: s_i = \begin{cases} s_i & \text{if } IoU(M, b_i) < N_t \\ s_i \times (1 - DIoU...结论 非极大值抑制(NMS)作为目标检测中的核心后处理技术,经历了从传统NMS到Soft-NMS、DIoU-NMS等变体的演进。每一次改进都针对传统NMS的局限性,提出了创新的解决方案。

    45010

    AAAI 2020 | DIoU和CIoU:IoU在目标检测中的正确打开方式

    论文提出了IoU-based的DIoU loss和CIoU loss,以及建议使用DIoU-NMS替换经典的NMS方法,充分地利用IoU的特性进行优化。...DIoU-NMS倾向于中心点距离较远的box存在不同的对象,而且仅需改几行代码,DIoU-NMS就能够很简单地集成到目标检测算法中 Experimental Results *** YOLO v3 on...[1240]   如图8所示,DIoU-NMS能够更好地引导bbox的消除 [1240]   为了进一步验证DIoU-NMS的效果,进行了对比实验。...如图9所示,DIoU-NMS的整体性能都比原生的NMS效果要好 Conclusion ***   论文提出了两种新的IoU-based损失函数,DIoU loss和CIoU loss:DIoU loss...另外论文提出DIoU-NMS来代替原生的NMS,充分地利用IoU的特性进行优化,从实验结果来看,效果也是很好的 写作不易,未经允许不得转载~ 更多内容请关注知乎专栏/微信公众号【晓飞的算法工程笔记】

    4.4K00

    YOLOv3 提升 5.91 mAP,IoU在目标检测中的正确打开方式

    论文提出了IoU-based的DIoU loss和CIoU loss,以及建议使用DIoU-NMS替换经典的NMS方法,充分地利用IoU的特性进行优化。...DIoU-NMS倾向于中心点距离较远的box存在不同的对象,而且仅需改几行代码,DIoU-NMS就能够很简单地集成到目标检测算法中 Experimental Results ---- YOLO v3 on...图7对GIoU和CIoU的结果进行了可视化,可以看到,在中大型物体检测上,CIoU的结果比GIoU要准确 Discussion on DIoU-NMS ?  ...如图8所示,DIoU-NMS能够更好地引导bbox的消除 ?   为了进一步验证DIoU-NMS的效果,进行了对比实验。...另外论文提出DIoU-NMS来代替原生的NMS,充分地利用IoU的特性进行优化,从实验结果来看,效果也是很好的。 ----

    2.4K10
    领券