基于轻量化神经网络的社交距离检测
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

陕西省科技计划重点项目(2017ZDCXL-GY-05-03)


Social Distance Detection Based on Lightweight Neural Network
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 增强出版
  • |
  • 文章评论
    摘要:

    保持安全社交距离是有效防止病毒传播的重要手段之一, 不仅可以减少感染者数量和医疗负担, 同时也极大降低死亡率. 在YOLOv4框架基础上使用轻量化网络E-GhostNet代替原网络中的CSPDarknet-53, E-GhostNet网络在输入数据和原始Ghost模块生成的输出特征之间建立关系, 使网络能够捕获上下文特征. 然后, 在E-GhostNet中引入坐标注意力机制(CA)增强模型对有效特征的关注. 另外, 使用SIoU损失函数更换CIoU损失获得更快的收敛速度和优化效果. 最后, 结合DeepSORT多目标跟踪算法来检测和标记行人, 并使用仿射变换(IPM)判定行人间距离的违规行为. 实验结果显示, 该网络检测速度为40 FPS, 精度值达到85.71%, 相比原始GhostNet算法提升2.57%, 达到实时行人距离检测的效果.

    Abstract:

    Maintaining a safe social distance is one of the important means to effectively prevent the spread of the virus. Moreover, it can not only reduce the number of infected people and ease the medical burden but also greatly lower the mortality rate. On the basis of the you only look once version 4 (YOLOv4) framework, the lightweight network E-GhostNet is used to replace the CSPDarknet-53 in the original network. The E-GhostNet network establishes a relationship between the input data and the output features generated by the original Ghost module, thereby enabling the network to capture contextual features. Then, the coordinate attention (CA) mechanism is introduced to E-GhostNet to enhance the model’s attention on effective features. In addition, the complete intersection over union (CIoU) loss function is replaced by the soft intersection over union (SIoU) loss function to obtain a faster convergence speed and optimization effect. Finally, the DeepSORT multi-target tracking algorithm is utilized to detect and label pedestrians, and affine transformation (IPM) is employed to determine the violation of the required distance between pedestrians. The experimental results show that the network achieves real-time pedestrian distance detection with a detection speed of 40 FPS and an accuracy of 85.71%, which is 2.57% higher than that of the original GhostNet algorithm.

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

王林,张江涛.基于轻量化神经网络的社交距离检测.计算机系统应用,2023,32(2):128-138

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2022-06-23
  • 最后修改日期:2022-07-25
  • 录用日期:
  • 在线发布日期: 2022-10-28
  • 出版日期:
您是第位访问者
版权所有:中国科学院软件研究所 京ICP备05046678号-3
地址:北京海淀区中关村南四街4号 中科院软件园区 7号楼305房间,邮政编码:100190
电话:010-62661041 传真: Email:csa (a) iscas.ac.cn
技术支持:北京勤云科技发展有限公司

京公网安备 11040202500063号