基于改进YOLOv5s的车辆行人检测
DOI:
CSTR:
作者:
作者单位:

1.天津理工大学;2.华北理工大学;3.广州计量检测技术研究院

作者简介:

通讯作者:

中图分类号:

TN2

基金项目:

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


Vehicle pedestrian detection based on improved YOLOv5s
Author:
Affiliation:

Fund Project:

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

    针对车辆行人检测中容易出现小目标错检、漏检的问题,提出了一种基于改进YOLOv5s的车辆行人检测算法,首先在不改变原有PANet的条件下,从头部网络引出第四个检测头,提高对小目标的检测能力,其次在主干网络中设计CF模块来代替原有的下采样模块,增强特征提取能力,然后将Neck网络中通道数减半的C3模块重新设计为S-C3,减少Neck部分的信息丢失问题,最后重新构建SPPF模块为D-SPP模块,为Neck部分保留更加细致的特征,提高对目标的检测精度。实验采用KITTI数据集,对数据集进行类别合并、删除等处理。实验表明,改进的算法与原算法在KITTI上相比对小目标错检、漏检的问题有明显改善。并且依旧满足实时性的要求。

    Abstract:

    Aiming at the problem that small targets are prone to misdetection and missing detection in vehicle pedestrian detection, a vehicle pedestrian detection algorithm based on improved YOLOv5s is proposed. First, a fourth detection head is extracted from the head network without changing the original PANet to improve the detection ability of small targets. Secondly, CF module is designed in the backbone network to replace the original subsampling module and enhance the feature extraction ability. Then C3 module, which has half the number of channels in the Neck network, is redesigned into S-C3 to reduce the information loss problem in the Neck part. Finally, SPPF module is rebuilt into D-SPP module to retain more detailed features for the Neck part. Improve the detection accuracy of the target. KITTI data set was used in the experiment to merge and delete the categories of the data set. The experimental results show that the improved algorithm is better than the original algorithm on KITTI for the problem of misdetection and missing detection of small targets. And still meet the requirements of real-time.

    参考文献
    相似文献
    引证文献
引用本文
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2023-07-13
  • 最后修改日期:2023-09-11
  • 录用日期:2023-09-18
  • 在线发布日期:
  • 出版日期:
文章二维码