Suppr超能文献

基于改进 YOLOv5 的口罩佩戴检测算法研究。

Research on Mask-Wearing Detection Algorithm Based on Improved YOLOv5.

机构信息

School of Mechanical Engineering, North China University of Water Resources and Electric Power, No. 36 Beihuan Road, Zhengzhou 450045, China.

出版信息

Sensors (Basel). 2022 Jun 29;22(13):4933. doi: 10.3390/s22134933.

Abstract

COVID-19 is highly contagious, and proper wearing of a mask can hinder the spread of the virus. However, complex factors in natural scenes, including occlusion, dense, and small-scale targets, frequently lead to target misdetection and missed detection. To address these issues, this paper proposes a YOLOv5-based mask-wearing detection algorithm, YOLOv5-CBD. Firstly, the Coordinate Attention mechanism is introduced into the feature fusion process to stress critical features and decrease the impact of redundant features after feature fusion. Then, the original feature pyramid network module in the feature fusion module was replaced with a weighted bidirectional feature pyramid network to achieve efficient bidirectional cross-scale connectivity and weighted feature fusion. Finally, we combined Distance Intersection over Union with Non-Maximum Suppression to improve the missed detection of overlapping targets. Experiments show that the average detection accuracy of the YOLOv5-CBD model is 96.7%-an improvement of 2.1% compared to the baseline model (YOLOv5).

摘要

新型冠状病毒(COVID-19)传染性极强,正确佩戴口罩可以阻碍病毒传播。然而,自然场景中存在遮挡、密集、小目标等复杂因素,常导致目标漏检和误检。针对这些问题,本文提出了一种基于 YOLOv5 的戴口罩检测算法 YOLOv5-CBD。首先,在特征融合过程中引入坐标注意力机制,突出关键特征,减少特征融合后冗余特征的影响。然后,用加权双向特征金字塔网络替换特征融合模块中原有的特征金字塔网络模块,实现高效的双向跨尺度连接和加权特征融合。最后,结合交并比和非极大值抑制来提高重叠目标的漏检率。实验表明,YOLOv5-CBD 模型的平均检测精度为 96.7%,比基线模型(YOLOv5)提高了 2.1%。

https://cdn.ncbi.nlm.nih.gov/pmc/blobs/193b/9269836/421480214b3f/sensors-22-04933-g001.jpg

文献AI研究员

20分钟写一篇综述,助力文献阅读效率提升50倍。

立即体验

用中文搜PubMed

大模型驱动的PubMed中文搜索引擎

马上搜索

文档翻译

学术文献翻译模型,支持多种主流文档格式。

立即体验