基于改进YOLOv8n的道路裂缝检测轻量化模型
DOI:
作者:
作者单位:

南京邮电大学物联网学院

作者简介:

通讯作者:

基金项目:

国家自然科学基金项目(面上项目,重点项目,重大项目)


A Lightweight Model of Road Crack Detection Based on Improved YOLOv8n
Author:
Affiliation:

Nanjing University of Posts and Telecommunications,Internet of Things College

Fund Project:

The National Natural Science Foundation of China (General Program, Key Program, Major Research Plan

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
    摘要:

    针对道路裂缝外观特征易受环境干扰、细小裂缝漏检率高、检测设备计算资源受限的问题,提出了轻量级检测模型MCA-YOLO-A。该模型基于YOLOv8n,将主干网络替换为更轻量的MobileNetV3特征提取网络,并融合了精确空间信息的CA注意力模块,提高了特征提取能力。同时,引入了适用于轻量级网络的Alpha-IOU损失函数,使得网络整体性能提升。此外,增加了小目标检测层,提升细小裂缝的识别精度。MCA-YOLO-A模型在道路裂缝数据集上平均精度均值mAP_0.5和F1分数分别达到0.930和0.893,相较于原YOLOv8n模型提升了7.0%和9.7%,参数量仅为6.0M,减少了4.8%,检测速度达到95帧/s。实验结果证明,该模型精度高、模型较小、泛化能力强,更适合部署在嵌入式系统和移动设备等计算资源受限的场景。

    Abstract:

    Aiming at the problems that the appearance characteristics of road cracks are easily disturbed by the environment, the missed detection rate of small cracks is high, and the computing resources of detection equipment are limited, a lightweight detection model MCA-YOLO-A is proposed. Based on YOLOv8n, this model replaces the backbone network with a lighter MobileNetV3 feature extraction network, and integrates the CA attention module with accurate spatial information to improve the feature extraction ability. At the same time, the Alpha-IOU loss function suitable for lightweight network is introduced, which improves the overall performance of the network. In addition, a small target detection layer is added to improve the identification accuracy of small cracks. The mAP_0.5 and F1 Score of MCA-YOLO-A model on road crack data sets is 0.930 and 0.893, respectively, which is 7.0% and 9.7% higher than that of the original YOLOv8n model, and the parameter quantity is only 6.0M, which is 4.8% lower, and the detection speed reaches 95 frames/s.. The experimental results show that the model has high accuracy, small model and strong generalization ability, and is more suitable for deployment in scenes with limited computing resources such as embedded systems and mobile devices.

    参考文献
    相似文献
    引证文献
引用本文
分享
文章指标
  • 点击次数:
  • 下载次数:
历史
  • 收稿日期:2024-08-08
  • 最后修改日期:2024-11-14
  • 录用日期:2025-01-10
  • 在线发布日期: