改进YOLO系列:12.Repulsion损失函数【遮挡】

本文深入探讨了针对物体遮挡问题的Repulsion损失函数,它由三部分组成,旨在提升YOLOv5的预测准确性。该损失函数能引导预测框靠近IOU最大的真实目标框,并与其他预测框保持距离。通过在YOLOv5中添加Repulsion损失,能有效解决密集遮挡物体的检测问题。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

1. RepLoss论文

物体遮挡问题可以分为类内遮挡和类间遮挡两种情况。类间遮挡产生于扎堆的同类物体,也被称为密集遮挡(crowd occlusion)。Repulsion损失函数由三个部分构成,yolov5样本匹配,得到的目标框和预测框-一对应第一部分主要作用:预测目标框吸引IOU最大的真实目标框,使得预测更加准确;
第二部分主要作用:远离除IOU最大值之外的最大的IOU目标框。根据预测目标框,寻找一个除了IOU最大的目标框外的次IOU最大目标框,不是根据目标框寻找一个次IOU最大的预测框,两个相反的概
念。第三部分主要作用:预测框之间互相远离;
论文题目:CCNet: Criss-Cross Attention for Semantic
Segmentation
论文链接:RepLoss

效果:
在这里插入图片描述

2. RepLoss原理

yolov5+Repulsion
在yolov5中使用Repulsion损失函数解决密集且遮挡的物体

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值