数据集
学生上课行为检测,我们直接使用公开数据集
共三类行为:举手、读书、写字
数据集已经按照YOLO格式配置好,数据内容如下
模型训练
采用YOLOv8模型进行训练,官方代码
首先是划分数据集,分为训练集、验证,已经划分好
后修改数据集配置文件: 需要修改数据集路径以及各个类别对应的名称
后修改训练代码中的配置:
主要修改数据集路径与预训练模型路径,我用的是yolov8-s.pt

训练效果
这些都是模型训练产生的结果,被放在了runs\train\exp 文件下
图像用户界面
网上可以可到很多开源界面,我这里提供了一个,放在了文件夹中
完整项目链接
因上传文件大小限制,数据集、训练代码、训练结果、UI界面都放在了百度网盘中,付费后即可获得下载链接