一种基于YOLOv5的口罩佩戴实时检测方法

摘要

本发明公开了一种基于YOLOv5的口罩佩戴实时检测方法，属于机器学习领域。该方法包括：制作用于口罩佩戴检测的数据集，分析YOLOv5模型用于口罩检测的问题，针对问题对YOLOv5网络进行了优化，采用K‑means++算法来计算anchor参数，并且对原先的损失函数进行修改，引入CIOU_Loss作为该算法损失函数的Bounding Box Regression Loss，然后对目标检测的后处理过程中引入DIOU_NMS来代替原来的加权非极大值抑制（NMS）。本发明算法提高了在人员密集处口罩检测的准确度，并且减少了因为遮挡而出现漏检的情况。