基于YOLOv5的教室人数检测统计系统
基于YOLOv5的教室人数检测统计系统可以有效地用于监控教室内的学生数量，适用于多种应用场景，比如
- 自动考勤、安全监控或空间利用分析
以下是如何构建这样一个系统的概述，包括环境准备、数据集创建、模型训练以及如何处理不同类型的媒体输入。
代码布置
.
├── dataset # 数据集
│ ├── annotations # XML 标签
│ ├── images # 图片
│ └── labels # txt 标签
├── detect.sh # detect shell 脚本
├── hellodata.py # 数据探索、预处理
├── LICENSE
├── README.md
├── res
│ ├── demo_picture1.png # 样例图片
│ └── demo_picture2.png
├── runs
│ ├── detect # 训练后的文件结果
│ └── train # 训练后的权重
├── train.sh # train shell 脚本
├── xml2txt.py # 将 XML 转换为 txt 标签
└── yolov5 # 从 @ultralytics/yolov5 clone
├── data
│ ├── coco.yaml
│ ├── headset.yaml # 自定义训练集
│ ├── hyp.scratch.yaml
│ ├── images # 存放 detect 输入数据
│ │ ├── bus.jpg
│ │ └── zidane.jpg
│ └── videos # 存放 detect 输入数据
├── detect.py
├── Dockerfile
├── hubconf.py
├── models # 预训练模型 YAML 文件
├── requirements.txt # Python 依赖库
├── test.py
├── train.py
├── utils
└── weights # 预训练权重
数据排布
目标检测 (Object Detection) 在教室人数统计上的应用尝试
先借助 YOLOv5 预训练模型对图片数据集进行训练，再测试多种输入流
数据集下载：Classroom Monitoring Dataset - kaggle
images 图片
partA 2000张，格式： [PartA_num].jpg
partB 2405张，格式：[PartB_num].jpg
annotations 标签，标注了图片中 目标的类别和坐标位置
partA 2000条，格式：[PartA_num].xml
partB 2405条，格式：[PartB_num].xml
主要工具包版本为 PyTorch 1.7.1+cu110 和 Python 3.8.5
代码运行
安装 Python 库依赖：pip3 install -r yolov5/requirements.txt
下载预训练权重，把下载的 .pt 文件部署在 yolov5/weights/ 路径下
代码训练
为了方便执行，编写了脚本 detect.sh 和 train.sh，分别进行训练与预测测试
根据训练效果调整 train.sh 文件中常用参数：
-epochs # 训练的 epoch，默认值 300
--batch-size # 默认值 16
--cfg yolov5s.yaml --weights '' # 从头开始训练
--cfg yolov5s.yaml --weights yolov5s.pt # 从预训练模型开始训练
--data # 数据集的配置文件，默认为 data/coco128.yaml
--resume # 是否从最新的 last.pt 中恢复训练，布尔值
--evolve # 进化超参数 (Evolve Hyperparameters)，布尔值
--cache-images # 缓存图片可以更快的开始训练，布尔值
--weights # 初始化参数路径，默认值 ''
--adam # 使用 adam 优化器，布尔值
一般只需改动这两个脚本文件就可，如需训练自定义的数据集，请参考官方文档：Train Custom Data - YOLOv5 Documentation
其他
自定义数据集
观察数据
数据集大小
数据集样本
图像分辨率
数据预处理
数据清洗，观察发现，有一些图像的 label 存在缺失，在 XML 转换 TXT 的过程中一并丢弃
YOLOv5 原生预处理
搭建模型，可视化分析
分析与调整训练，提高模型泛化能力
结果
多视角 稳定识别！
demo 的 detect 结果：
最后
计算机视觉、图像处理、毕业辅导、作业帮助、代码获取，远程协助，代码定制，私聊会回复!
原文地址：https://blog.csdn.net/ALiLiLiYa/article/details/142385635
免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！