作者投稿和查稿 主编审稿 专家审稿 编委审稿 远程编辑

计算机工程 ›› 2022, Vol. 48 ›› Issue (10): 28-36. doi: 10.19678/j.issn.1000-3428.0063344

• 热点与综述 • 上一篇    下一篇

COVID‐19疫情下基于YOLOv4的安全社交距离风险评估

郭克友1, 贺成博1, 王凯迪1, 王苏东1, 李雪1, 张沫2   

  1. 1. 北京工商大学 人工智能学院, 北京 100048;
    2. 交通运输部公路科学研究院, 北京 100088
  • 收稿日期:2021-11-24 修回日期:2021-12-31 发布日期:2022-07-04
  • 作者简介:郭克友(1975—),男,副教授,主研方向为嵌入式开发、机器视觉;贺成博、王凯迪、王苏东、李雪,硕士研究生;张沫,研究员。
  • 基金资助:
    交通运输行业重点科技项目“面向自动驾驶汽车封闭测试环境的可遥控仿真行人测试装备研发”(2018-ZD1-010);北京工商大学2021年研究生科研能力提升计划项目。

Risk Assessment of Safety Social Distance Based on YOLOv4 in the COVID-19 Pandemic

GUO Keyou1, HE Chengbo1, WANG Kaidi1, WANG Sudong1, LI Xue1, ZHANG Mo2   

  1. 1. School of Artificial Intelligence, Beijing Technology and Business University, Beijing 100048, China;
    2. Institute of Highway Science, Ministry of Transport, Beijing 100088, China
  • Received:2021-11-24 Revised:2021-12-31 Published:2022-07-04

摘要: 为保持行人在新型冠状病毒肺炎(COVID-19)疫情下的安全社交距离,有效控制和预防疫情传播,构建一种基于YOLOv4的安全社交距离风险评估模型。利用微调后的YOLOv4算法对行人进行目标提取,获取行人关键点,并将行人连续运动视为质点的连续运动,结合DeepSort算法实现对行人的跟踪处理。在此基础上,建立视觉坐标系,在鸟瞰视角下提出运动矢量分析算法计算和判断行人运动方向并评估行人的安全社交距离。在牛津城市中心的数据集上评估模型有效性,实验结果表明,微调后YOLOv4算法在行人检测中平均精度均值达到90.33%,行人社交距离风险评估准确率达到88.23%,性能优于Fast R-CNN、Faster R-CNN、YOLOv3和YOLOv4算法,表明所提模型能够有效提升安全社交距离的检测准确性。

关键词: YOLO网络, 安全社交距离, 新型冠状病毒肺炎, 目标检测, 运动矢量分析

Abstract: In order to maintain a safety social distance in the COVID-19 pandemic, control and prevent pandemic transmission, a risk assessment model of safety social distance based on YOLOv4 is constructed.The fine-tuned YOLOv4 algorithm extracts the pedestrian target, obtains their key points, and regards their continuous movement as a continuous movement of particles.When combined with the DeepSort algorithm, it can be used to track pedestrians.On this basis, a visual coordinate system is established, and a motion vector analysis algorithm is proposed to calculate and judge the direction of pedestrian movement and evaluate the safety social distance of pedestrians from an aerial perspective.The effectiveness of the model is evaluated on the Oxford City Center Dataset.The experimental results show that the mean Average Precision(mAP) of the fine-tuned YOLOv4 algorithm in pedestrian detection reaches 90.33%, and the accuracy of pedestrian social distance risk assessment reaches 88.23%.These values are superior to those of the Fast R-CNN, Faster R-CNN, YOLOv3, and YOLOv4 algorithms.Thus, the proposed model effectively improves the detection accuracy of safe social distance.

Key words: YOLO network, safety social distance, Corona Virus Disease 2019(COVID-19), object detection, motion vector analysis

中图分类号: