改进MobileViT与YOLOv4的轻量化车辆检测网络
DOI:
作者:
作者单位:

新疆大学电气工程学院 乌鲁木齐 830017

作者简介:

通讯作者:

中图分类号:

TP391

基金项目:

国家自然科学基金(51468062)项目资助


Lightweight vehicle detection network based on MobileViT and YOLOv4
Author:
Affiliation:

School of Electrical Engineering, Xinjiang University,Urumqi 830017, China

Fund Project:

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

    基于深度学习的目标检测算法在智能交通的应用中,对于车辆检测存在模型参数量大、计算速度慢和简单网络精准度较低的问题。本文提出了一种高效的轻量化车辆检测模型,该检测模型采用YOLOv4网络作为参考模型进行改进。首先,本文采用CSPMobileViT网络来替换原始主干网络,然后将PANet替换成BiFPN,并且将BiFPN中的3×3标准卷积替换成深度可分离卷积,最后,在BiFPN之前和YOLO-Head之前添加ECA模块。在损失函数部分,将边框回归损失CIoU改进为Focal EIoU来解决难易样本不平衡的问题。实验结果表明改进网络的mAP值为96.77%,检测速度达到每张图片0.023 4 s,模型大小只有32.76 MB,参数量为8 587 541,与原始算法相比mAP提升了1.54%,而模型大小和参数量仅约为原始模型1/8,并且FPS提升了7.5,改进算法具有更好检测效果。

    Abstract:

    In the application of Intelligent Transportation, target detection algorithm based on deep learning has the problems of large number of model parameters, slow calculation speed and low accuracy of simple network for vehicle detection. This paper presents an efficient lightweight vehicle detection model, which is improved by using YOLOv4 network as a reference model. First, this paper uses CSPMobileViT network to replace the original backbone network, then replaces PANet with BiFPN, replaces 3×3 standard convolution in BiFPN with deep detachable convolution, and finally adds ECA module before BiFPN and YOLO-Head. In the loss function section, the Border Regression Loss CIoU is improved to Focal EIoU to solve the problem of difficult sample imbalance. The experimental results show that the mAP value of the improved network is 96.77%, the detection speed reaches 0.023 4 s per picture, the model size is only 32.76 MB, and the parameter amount is 8 587 541. Compared with the original algorithm, the mAP is improved by 1.54%, while the model size and number of parameters are only about 1/8 of the original model, and the FPS is improved by 7.5, so the improved algorithm has better detection effect.

    参考文献
    相似文献
    引证文献
引用本文

郑玉珩,黄德启.改进MobileViT与YOLOv4的轻量化车辆检测网络[J].电子测量技术,2023,46(2):175-183

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-03-11
  • 出版日期: