基于改进YOLOv5模型的复杂场景口罩佩戴识别

来源 :微处理机 | 被引量 : 0次 | 上传用户:lzxs123
下载到本地 , 更方便阅读
声明 : 本文档内容版权归属内容提供方 , 如果您对本文有版权争议 , 可与客服联系进行内容授权或下架
论文部分内容阅读
为满足疫情时期的特殊需要,基于改进YOLOv5模型,设计一款应用于复杂场景的口罩佩戴识别检测系统.系统采用PyCharm集成开发环境,从网络上爬取1600张口罩佩戴相关的图片,在原始k-means算法基础上加入聚类算法,获取与真实框之间的更高的先验框.在人机交互界面使用Qt组件设计,图像和模型数据加载使用开源OpenCV视觉库实现.口罩佩戴检测的核心算法使用目标检测算法中的YOLOv5训练并结合PyTorch框架实现.实验结果表明,系统能够实现多人场景下人群佩戴口罩状况的检测,在多人物以及实时检测场景中准确率有所提高,在使用和效果上都具有自身特有的优势.
其他文献