R-YOLO轨道人员目标检测模型
CSTR:
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

河北省自然科学基金(F2022208002);河北省高等学校科学技术研究重点项目(ZD2021048)


R-YOLO orbital personnel target detection model
Author:
Affiliation:

Fund Project:

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

    针对现有铁路人员入侵识别准确率不高、实时性较差的问题,在YOLOv4模型的基础上提出一种R-YOLO轨道人员目标检测模型。首先,用轻量级骨干网络ResNet50代替原有的CSPDarknet53网络,利用深度可分离卷积替代PANet中的标准卷积,减少网络层数以及模型体积,加快模型的识别速度。其次,在加强特征提取网络的3个特征层分别加入有效通道注意力模块,采用K-means++聚类算法重新对数据集进行聚类和分析,提高目标检测模型的精度;在模型训练方面,采用迁移学习和混合数据集联合训练,解决人员识别精度不理想以及误检漏检等问题。最后,利用R-YOLO轨道人员目标检测模型对真实铁路人员入侵数据集进行测试。结果表明,R-YOLO模型在真实铁路人员入侵数据集上的平均识别精度达到了92.12%,较传统YOLOv4算法高出1.89%,帧速率由38.74 f·s-1提升到47.73 f·s-1。R-YOLO模型部分解决了铁路入侵人员误检漏检问题,提高了铁路人员入侵识别的实时性和准确率,为铁路安全运行提供了保障。

    Abstract:

    Aiming at the problems of low accuracy and poor real-time recognition of railway personnel intrusion, an R-YOLO track personnel target detection model was proposed based on the YOLOv4 model. Firstly, the original CSPDarknet53 network was replaced by the lightweight backbone network ResNet50, and the standard convolution in PANet was replaced by deep separable convolution, which reduces the number of network layers and model volume, and accelerates the model recognition speed. Secondly, the effective channel attention module was added to the three feature layers before and after strengthening the feature extraction network, and the K-means++ clustering algorithm was used to re-cluster and analyze the dataset to improve the accuracy of the target detection model. In terms of model training, transfer learning and hybrid dataset joint training were used to solve the problems of poor personnel identification accuracy,false detection and leakage detection. Finally, the R-YOLO track personnel target detection model was used to test the real railway personnel intrusion dataset. The experimental results show that the average recognition accuracy of the R-YOLO model on the real railway personnel intrusion dataset reaches 92 .12%, which is 1 .89% higher than that of the traditional YOLOv4 algorithm, and the frame rate increases from 38 .74 f·s-1 to 47 .73 f·s-1. The R-YOLO model improves the problem of false detection and leakage of railway intrusion personnel, improves the real-time and accuracy of railway personnel intrusion identification, and provides guarantee for the safe operation of railway.

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

张永强,李胜男,张子强,刘健章,张 坤,苗 磊. R-YOLO轨道人员目标检测模型[J].河北科技大学学报,2023,44(6):580-588

复制
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2023-04-21
  • 最后修改日期:2023-10-30
  • 录用日期:
  • 在线发布日期: 2023-12-30
  • 出版日期:
文章二维码