Infra-YOLO:高效神经网络结构与模型压缩用于实时红外小物体检测
原文中文,约300字,阅读约需1分钟。发表于: 。本研究针对红外小物体检测中的低信噪比和数据集缺乏问题,构建了新数据集InfraTiny,并提出了多尺度注意机制模块和特征融合增强金字塔模块。通过在YOLO模型中集成这些方法,Infra-YOLO的红外小物体检测性能显著提升,特别是在嵌入式设备上的应用表现卓越,展示了该方法在速度与精度之间的有效平衡。
本研究构建了新数据集InfraTiny,解决了红外小物体检测中的低信噪比和数据集缺乏问题。通过集成多尺度注意机制模块和特征融合增强金字塔模块到YOLO模型中,Infra-YOLO的红外小物体检测性能显著提升,尤其在嵌入式设备上表现出色,实现了速度与精度的有效平衡。