专栏介绍:YOLOv9改进系列 | 包含深度学习最新创新,助力高效涨点!!!
一、Innor-iou介绍
作者导读:https://zhuanlan.zhihu.com/p/666039955
代码链接:https://github.com/malagoutou/Inner-IoU
论文链接:https://arxiv.org/abs/2311.02877
二、Innor-iou模块详解
2.1 论文摘要
随着检测器的迅速发展, 边框回归取得了巨大的进步。然而,现有的基于 IoU 的边框回归仍聚焦在通过加入新的损失项来加速收敛,忽视 IoU 损失项其自身的限制。尽管理论上 IoU 损失能够有效描述边框回归状态,在实际应用中,它无法根据不同检测器与检测任务进行自我调整,不具有很强的泛化性。基于以上,我们首先分析了 BBR 模式,得出结论在回归过程区分不同回归样本并且使用不同尺度的辅助边框计算损失能够有效加速边框回归过程。对于高 IoU 样本,使用较小的辅助边框计算损失能够加速收敛,而较大辅助边框适用于低 IoU 样本。接着,我们提出了 Inner-IoU Loss, 其通过辅助边框计算 IoU 损失。针对不同的数据集与检测器,我们引入尺度因子 ratio 控制辅助边框的尺度大小用于计算损失。最后,将 Inner-IoU 集成至现有的基于 IoU 损失函数中进行仿真实验与对比实验。实验结果表明在使用本文所提出方法后检测效果得到进一步提升,验证了本文方法的有效性以及泛化能力。
三、 HGBlock模块使用教程
3.1 HGBlock模块的代码
3.2 在YOLO v9中的添加教程
3.3 运行配置文件
这部分文章暂不开源!
⭐YOLOv9改进创新来啦!最新的YOLO系列模型,发论文必备!⭐
⭐最新推出YOLOv9创新点项目,目前已有30+创新!⭐
⭐后期更新包含模块、卷积、检测头、损失等改进!⭐
⭐四月底预计创新点可达80-100+!⭐
⭐现在入手仅$ 59.9,早入早发论文!⭐
⭐即将涨价,创新点越多,价格越贵!⭐
联系QQ: 2668825911 ,欢迎交流!
本项目持续更新,不付费订阅也可关注等每周更新,每周开源1-2篇。
标签:辅助,iou,损失,边框,Inner,IoU,YOLOv9 From: https://blog.csdn.net/StopAndGoyyy/article/details/136843044