SSD没有错付 | SSD改进版本来啦!PSSD携实时性与高精度归来!(二)

简介: SSD没有错付 | SSD改进版本来啦!PSSD携实时性与高精度归来!(二)

3、Loss Settings based on IOU


3.1、IOU branch

分类和回归任务相对独立,分类置信度不能准确地表示预测框的位置。因此,NMS会过滤一些更准确的预测框。作者使用一种成本较低的独立于类别的方法,对NMS中使用的分数进行基于IOU的衰减。

image.png

如图6所示,添加了一个分支来预测回归框和它们的GT之间的IOU。结果是通过在增强的特征图上直接应用3×3卷积,然后使用sigmoid函数将其归一化为0-1得到的。计算回归框与GT之间的IOU。

为了确保损失的凸性并增加梯度值,设计了一种新的IOU回归损失(),如下所示:

image.png

其中是预测的并集上的交集,是并集上目标交集,分别是IOU和目标IOU的精度。

3.2、Regression Loss

为了使大目标更好地回归,使用由Libra-RCNN提出的balance  ():

image.png

其中,默认参数设置为α=0.5和γ=1.5。

3.3、Cross-entropy of joint IOU loss:

对于一般的单阶段检测算法,如RetinaNet,IOU在0.4-0.5范围内的样本被忽略,因为模型很难学习分类边界。然而,仍然存在以下问题:

  • 存在一些假阳性样本(回归结果的IOU小于0.5),模型仍然迫使它们被识别为吗,这导致低质量预测框。
  • 有些样本的IOU在0.4-0.5之间,可以学习,忽略这些样本不利于提高模型的学习能力。
  • 传统的交叉熵损失没有考虑推理过程中NMS中使用的分数生成。

因此,采用了新的正样本规则,并设计了一个新的联合IOU()交叉熵损失,如下所示:

image.png

其中是梯度值,是预测类概率。将其GT值大于0.4的IOU的默认框设置为正样本。如果可以更准确地回归默认框(≥ 0.5),计算分类损失,否则忽略它们。

因此,该模型不强制对假阳性样本进行分类,可以挖掘出更多潜在的阳性样本。同时,将最终优化目标更改为最大化·,这与推断一致(得分=·)。此外,考虑到平衡未从回归的整体进行优化该损失函数可以将关于IOU的监测信号添加到回归器,以使预测结果更加准确。


4、实验


4.1、Performance evaluation on Microsoft COCO

image.png

4.2、Ablation Study

image.png

4.3、FEM module

image.png

4.4、Performance evaluation on Pascal VOC 2007


5、参考


[1].Precise Single-stage Detector.


6、推荐阅读


全新语义分割方法SegViT | 沈春华老师团队提出全新语义分割方法

目标检测技能点|一步一步带你实验并重新思考目标检测不同检测头的作用

必备技能 | YOLOv6中的用Channel-wise Distillation进行的量化感知训练来自哪里?

相关文章
|
SQL 算法 Java
【其他】快出数量级的性能是怎样炼成的
【其他】快出数量级的性能是怎样炼成的
51 0
路径损耗计算模型 | 带你读《大规模天线波束赋形技术原理与设计 》之二十五
本小节介绍 3D 信道的路损模型,是以 ITU 信道为基础拓展得到的。
12783 0
路径损耗计算模型  | 带你读《大规模天线波束赋形技术原理与设计 》之二十五
3GPP相应的5G UDN场景及性能需求 | 带你读《5G UDN(超密集网络)技术详解》之十九
相比 IMT-2020 推进组,3GPP 对 5G UDN 场景的研究更加具有针对性。 在业务应用独立式部署(SA)技术组的研究中,与 UDN 相关的场景都是服务 室内用户的,具体有两种:办公室和热点区域。
3GPP相应的5G UDN场景及性能需求 | 带你读《5G UDN(超密集网络)技术详解》之十九
|
3月前
|
SQL 算法 Java
快出数量级的性能是怎样炼成的
我们通过使用开源 SPL 重写了多个金融行业的 SQL 任务,实现了显著的性能提升,如保险公司团保明细单查询提速 2000+ 倍、银行 POS 机交易报表提速 30+ 倍等。这些优化的核心在于使用了更低复杂度的算法,而非依赖硬件加速。SPL 基于离散数据集理论,提供了丰富的高性能算法,使得复杂任务的优化成为可能。更多案例和详细技术解析可参见乾学院的相关课程和图书。
|
机器学习/深度学习 固态存储 算法
SSD没有错付 | SSD改进版本来啦!PSSD携实时性与高精度归来!(一)
SSD没有错付 | SSD改进版本来啦!PSSD携实时性与高精度归来!(一)
194 0
|
机器学习/深度学习 编解码 固态存储
暴力改进SSD | 小目标检测的福音
暴力改进SSD | 小目标检测的福音
283 0
|
存储 缓存 人工智能
PLC SSD虽来但远,QLC SSD火力全开
PLC NAND的普及还需要一定的时日。乐观估计,PLC SSD最快也要到2025年才可能量产进入市场。所以,PLC SSD还是一个未来的产品,当前我们的目光还是聚集在QLC身上吧。
|
机器学习/深度学习 固态存储 算法
SSD没有错付 | SSD改进版本来啦!PSSD携实时性与高精度归来!
SSD没有错付 | SSD改进版本来啦!PSSD携实时性与高精度归来!
400 0
SSD没有错付 | SSD改进版本来啦!PSSD携实时性与高精度归来!
|
存储 缓存 固态存储
面对SSD的步步紧逼,HDD依然奋斗不息
固态硬盘的步步紧逼,机械硬盘厂商也不会坐以待毙等着被SSD取代。HDD的厂商也在不断的创新,增加HDD的竞争优势。
|
存储 监控 芯片
稳定性差的硬伤 固态硬盘无法取代机械硬盘的原因
稳定性差的硬伤 固态硬盘无法取代机械硬盘的原因
443 1
稳定性差的硬伤 固态硬盘无法取代机械硬盘的原因