基于YOLOv5的口罩佩戴实时检测系统

基于YOLOv5的口罩佩戴实时监测系统

概述

基于YOLOv5的口罩检测系统，使用Flask进行开发，将检测压力分到各客户端侧进行端侧实时检测，对服务端性能需求小，可以部署在各种服务器甚至树莓派上。mAP达到0.96，检测结果较为准确。
本项目可以实现对实时网络视频流以及 本地摄像头的拉取检测，并实现了实时人群口罩佩戴率计算展示和统计，可以对统计图表结果进行下载存储。

工作流程

互联网搜集图片，筛选并手动标注获取数据集。
基于YOLOv5s训练模型
用Flask框架进行检测平台前后端开发
服务端部署在树莓派或其他服务器进行调试运行

开发概述

开发环境：Windows11 22H2 64bit
IDE：PyCharm
使用语言： Python、JavaScript
模型训练环境: NVIDIA V100 虚拟化 CPU 2核 内存 16G / CPU

项目效果

数据集概况

模型性能

环境配置

YOLOv5官方项目拉取

依赖及其版本：

(tfjs) D:\yolov5>conda list
# packages in environment at D:\Software\Anaconda\envs\tfjs:
# Name                    Version                   Build  Channel
numpy                     1.24.1                   pypi_0    pypi
pandas                    1.5.3                    pypi_0    pypi
pillow                    9.4.0                    pypi_0    pypi
python                    3.8.5                h5fd99cc_1    defaults
seaborn                   0.12.2                   pypi_0    pypi
tensorboard               2.9.0                    pypi_0    pypi
tensorboard-data-server   0.6.1