轨道交通视频中乘客口罩佩戴检测算法研究
李永玲1,2秦勇1曹志威1,2谢征宇2吴志宇1,3
Detection Algorithm of Passengers Wearing Masks in Rail Transit Video
LI YonglingQIN YongCAO ZhiweiXIE ZhengyuWU Zhiyu
-
作者信息:1.1.北京交通大学轨道交通控制与安全国家重点实验室， 100044， 北京；
2.2.北京交通大学交通运输学院， 100044， 北京；
3.3.北京交通大学软件学院， 100044， 北京
-
Affiliation:State Key Laboratory of Rail Traffic Control and Safety, Beijing Jiaotong University, 100044, Beijing, China
-
关键词:
-
Key words:
-
DOI:10.16037/j.1007-869x.2022.12.014
-
中图分类号/CLCN:F530.7;R126.4
-
栏目/Col:研究报告
摘要:
为应对新冠疫情下乘客乘坐轨道交通时必须佩戴口罩的情况，提出一种基于深度学习的轻量化口罩检测算法（Mask-Det算法）。首先，使用轻量化骨干网络EfficientNet提取图像特征；然后，利用高效的特征融合模块增强用于检测小目标的浅层特征图的语义信息；最后，算法在公共场景数据集上训练，并使用迁移学习在轨道交通数据集上做进一步优化。Mask-Det算法检测准确率高、模型参数小、检测速度快,可以实时检测各场所乘客是否佩戴口罩，有效减轻工作人员压力，提高进站速度。
Abstracts:
In response to the mandatory situation of passengers wearing masks on rail transit during COVID-19 pandemic period, a lightweight mask detection algorithm (Mask-Det) based on deep learning is proposed. First, the lightweight backbone network EfficientNet is used to extract image features. Then, a highly efficient feature fusion module is used to enhance the semantic information of the shallow feature map for detecting small targets. Finally, the algorithm is trained on the dataset of public scenarios, and then further optimized on the dataset of rail transit scenarios using transfer learning. Mask-Det algorithm has high detection accuracy, small model parameters, and fast detection speed. The algorithm can detect in real time whether passengers are wearing masks at various places, thus effectively alleviate personnel stress and improve passenger entry speed.
- 上一篇: 高铁物流运输模式及其可行性
- 下一篇: 数字赋能:中车唐山公司数字化转型之路
过刊浏览
资料下载
友情链接