轻量化红外目标检测模型,兼顾精度与实时性。
MS-YOLO: Infrared Object Detection for Edge Deployment via MobileNetV4 and SlideLoss

- 用MobileNetV4替代原骨干网络,降低计算开销。
- 引入SlideLoss提升小目标和遮挡样本检测精度。
- 在FLIR ADAS V2上实现6.7 GFLOPs的高效部署。
红外成像在低光及恶劣天气条件下具备优于可见光相机的城市物体检测能力。然而,类别不平衡、热噪声和计算资源限制会显著影响实际应用中的模型表现。本文在FLIR ADAS V2数据集上评估多个YOLO变体,最终以YOLOv8为基线。在此基础上提出MS-YOLO(基于MobileNetV4与SlideLoss的YOLO),将原模型的CSPDarknet骨干替换为更高效的MobileNetV4,使计算量减少1.5%且保持高精度。同时引入SlideLoss,动态增强少数类与遮挡样本的损失权重,提升检测精度而不牺牲召回率。在FLIR ADAS V2基准测试中,MS-YOLO达到具有竞争力的mAP,且推理仅需6.7 GFLOPs,验证了其在保证检测质量的同时有效降低计算成本,适用于城市环境下的实时边缘部署。
原文摘要 · Abstract (English)
Infrared imaging has emerged as a robust solution for urban object detection under low-light and adverse weather conditions, offering significant advantages over traditional visible-light cameras. However, challenges such as class imbalance, thermal noise, and computational constraints can significantly hinder model performance in practical settings. To address these issues, we evaluate multiple YOLO variants on the FLIR ADAS V2 dataset, ultimately selecting YOLOv8 as our baseline due to its balanced accuracy and efficiency. Building on this foundation, we present \texttt{MS-YOLO} (\textbf{M}obileNetv4 and \textbf{S}lideLoss based on YOLO), which replaces YOLOv8's CSPDarknet backbone with the more efficient MobileNetV4, reducing computational overhead by \textbf{1.5%} while sustaining high accuracy. In addition, we introduce \emph{SlideLoss}, a novel loss function that dynamically emphasizes under-represented and occluded samples, boosting precision without sacrificing recall. Experiments on the FLIR ADAS V2 benchmark show that \texttt{MS-YOLO} attains competitive mAP and superior precision while operating at only \textbf{6.7 GFLOPs}. These results demonstrate that \texttt{MS-YOLO} effectively addresses the dual challenge of maintaining high detection quality while minimizing computational costs, making it well-suited for real-time edge deployment in urban environments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。