首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

【目标检测】 开源 | CVPR2020 | 将DIoU和CIoU Loss用于目标检测的Bbox回归,表现SOTA

见文章底部 下载完整原文,公众号回复:1911.08287 论文地址:http://arxiv.org/pdf/1911.08287v1.pdf 代码:https://github.com/zzh-tju/diou...来源:天津大学 论文名称:Distance-IoU Loss: Faster and Better Learning for Bounding BoxRegression 原文作者:Zhaohui Zheng...本文提出了一个Distance-IoU (DIoU) loss,合并了预测框和目标框之间的标准化距离,在训练中比IoU和GIoU loss收敛得更快。...通过将DIoU和CIoU损失合并到YOLOv3、SSD和Faster RCNN等最新的目标检测算法,在IoU度量方面和GIoU度量方面实现了显著的性能提高。...而且DIoU很容易地应用到非最大抑制(non-maximum suppression,NMS)作为准则,进一步促进性能提升。 下面是论文具体框架结构以及实验结果: ? ? ? ? ? ? ?

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

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

论文提出了IoU-based的DIoU loss和CIoU loss,以及建议使用DIoU-NMS替换经典的NMS方法,充分地利用IoU的特性进行优化。...DIoU loss的惩罚项能够直接最小化中心点间的距离,而GIoU loss意在减少外界包围框的面积   DIoU loss保留了IoU loss和GIoU loss的一些属性: DIoU loss依然是尺寸无关的...DIoU loss也有一些独有的属性: 如图1和图3所示,DIoU loss能够直接最小化bbox的中心点距离。...最后,CIoU loss的梯度类似于DIoU loss,但还要考虑的梯度。...如图9所示,DIoU-NMS的整体性能都比原生的NMS效果要好 Conclusion ----   论文提出了两种新的IoU-based损失函数,DIoU loss和CIoU lossDIoU loss

2.1K10

基于DIou改进的YOLOv3目标检测

直观的展示如下图所示: DIoU loss的惩罚项能够直接最小化中心点间的距离,而GIoU loss意在减少外界包围框的面积 DIoU loss保留了IoU loss和GIoU loss的一些优点: DIoU...loss与尺寸无关的,不会大尺寸产生大的loss,小尺寸产生小的loss那样; 类似于GIoU lossDIoU loss能够为无交集的bbox提供梯度方向; 当bbox完全重合时, ,当bbox...在模拟实验中,发现DIoU loss也有一些独有的特点: DIoU loss可以直接最小化两个目标框的距离,因此比GIoU loss收敛快得多; 对于包含两个框在水平方向和垂直方向上这种情况,DIoU损失可以使回归非常快...由 定义可以看出来,损失函数会更加倾向于往重叠区域增多方向优化: 最后,CIoU loss的梯度类似于DIoU loss,但还要考虑 的梯度。...总结 论文提出了两种新的IoU-based损失函数,DIoU loss和CIoU lossDIoU loss最小化bbox间的中心点距离,从而使得函数快速收敛;CIoU loss则在DIoU loss

47510

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

论文提出了IoU-based的DIoU loss和CIoU loss,以及建议使用DIoU-NMS替换经典的NMS方法,充分地利用IoU的特性进行优化。...一般地,GIoU loss不能很好地收敛SOTA算法,反而造成不好的结果 DIoU   综合上面的分析,论文提出Distance-IoU(DIoU) loss,简单地在IoU loss基础上添加一个惩罚项...,而GIoU loss意在减少外界包围框的面积   DIoU loss保留了IoU loss和GIoU loss的一些属性: DIoU loss依然是尺寸无关的,不会大尺寸产生大的loss,小尺寸产生小的...的梯度类似于DIoU loss,但还要考虑$v$的梯度。...如图9所示,DIoU-NMS的整体性能都比原生的NMS效果要好 Conclusion ***   论文提出了两种新的IoU-based损失函数,DIoU loss和CIoU lossDIoU loss

3.8K00

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

DIoU loss的惩罚项能够直接最小化中心点间的距离,而GIoU loss意在减少外界包围框的面积 DIoU loss保留了IoU loss和GIoU loss的一些优点: DIoU loss与尺寸无关的...,不会大尺寸产生大的loss,小尺寸产生小的loss那样; 类似于GIoU lossDIoU loss能够为无交集的bbox提供梯度方向; 当bbox完全重合时, ?...在模拟实验中,发现DIoU loss也有一些独有的特点: DIoU loss可以直接最小化两个目标框的距离,因此比GIoU loss收敛快得多; 对于包含两个框在水平方向和垂直方向上这种情况,DIoU损失可以使回归非常快...Complete IoU loss DIoU Loss 只是考虑了边界框的重叠面积和 ? 与 ? 中心点距离,然而anchor框和目标框之间的w,h比的一致性也是极其重要的。...总结 论文提出了两种新的IoU-based损失函数,DIoU loss和CIoU lossDIoU loss最小化bbox间的中心点距离,从而使得函数快速收敛;CIoU loss则在DIoU loss

1.6K20

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

4、DIoU 综合上面的分析,论文提出Distance-IoU(DIoU) loss,简单地在IoU loss基础上添加一个惩罚项,该惩罚项用于最小化两个bbox的中心点距离。...DIoU loss的完全定义如公式7。...图 5 DIoU loss的惩罚项能够直接最小化中心点间的距离,而GIoU loss意在减少外界包围框的面积   DIoU loss保留了IoU loss和GIoU loss的一些属性: DIoU loss...在模拟实验中,发现DIoU loss也有一些独有的属性: 如图1和图3所示,DIoU loss能够直接最小化bbox的中心点距离。...五、总结 论文提出了两种新的IoU-based损失函数,DIoU loss和CIoU lossDIoU loss最小化bbox间的中心点距离,从而使得函数快速收敛;CIoU loss则在DIoU loss

2.7K30

类别不均衡问题之loss大集合:focal loss, GHM loss, dice loss 等等

最近在工作中也是碰到这个问题,花了些时间梳理并实践了类别不均衡问题的解决方式,主要实践了“魔改”loss(focal loss, GHM loss, dice loss 等),整理如下。...因此,就集中看下重加权 loss 改进的部分。 2. 模型层面的重加权 重加权主要指的是在 loss 计算阶段,通过设计 loss,调整类别的权值对 loss 的贡献。...比较经典的 loss 改进应该是 Focal Loss, GHM Loss, Dice Loss。...在交叉熵 loss 基础上,当正样本预测值  大于 0.5 时,需要给它的 loss 一个小的权重值 ,使其对总 loss 影响小,反之正样本预测值  小于 0.5,给它的 loss 一个大的权重值。...& DSC Loss Dice Loss 是来自文章 V-Net 提出的,DSC Loss 是香侬科技的 Dice Loss for Data-imbalanced NLP Tasks。

2.9K30

Contrastive Loss(对比损失)Contrastive Loss

Contrastive Loss 在传统的siamese network中一般使用Contrastive Loss作为损失函数,这种损失函数可以有效的处理孪生神经网络中的paired data的关系。...siamese network-孪生神经网络 contrastive loss的表达式如下: ?...# tensorflow伪代码 def contrastive_loss(self, y,d,batch_size): tmp= y *tf.square(d) #tmp= tf.mul...观察上述的contrastive loss的表达式可以发现,这种损失函数可以很好的表达成对样本的匹配程度,也能够很好用于训练提取特征的模型。当y=1(即样本相似)时,损失函数只剩下 ?...与欧式距离之间的关系,因为欧式距离越大与"两个文本相似"这个事实事与愿违,那么我们就对它们惩罚越大(即loss越大)。

3.9K40

Yolo发展史(v4v5的创新点汇总!)

DIOU Loss 和 CIOU Loss 好的目标框回归函数应该考虑三个重要几何因素:重叠面积、中心点距离,长宽比。...但是DIOU Loss没有考虑长宽比,当预测框在目标框内部时,且多个预测框的中心点的位置都一样时,这时候DIOU Loss无法区分这几个预测框的位置。 ?...因为DIOU在计算loss的时候,需要考虑边界框中心点的位置信息,所以一些重叠物体也可以回归出来。因此在重叠目标的检测中,DIOU_nms的效果优于传统的nms。...CIOU Loss的性能要比DIOU Loss好,那为什么不用CIOU_nms,而用DIOU_nms?...所以Yolov4在DIOU_Loss的基础上采用DIOU_nms的方式,而Yolov5中采用加权nms的方式。 采用DIOU_nms,对于遮挡问题,检出效果有所提升。

1.9K30
领券