【大作业-36】人眼瞳孔检测模型,YOLOv5/v8/11/12+PyQt+MySQL
本文介绍了基于YOLO系列模型的人眼瞳孔检测系统,包含YOLOv5/v8/11/12和RT-DETR五种模型的对比实验。系统采用PyQt设计交互界面,支持图片、视频、摄像头等多种检测方式,并集成MySQL数据库实现用户登录注册功能。实验数据显示YOLOv8s在精度和速度上表现最优。项目提供完整代码资源(49.9元)及详细配置指南,包括数据集准备、Python环境搭建和MySQL数据库配置等内容,适
B站:宋大水
人眼瞳孔检测,YOLOv5/v8/11/12+PyQt+MySQL_哔哩哔哩_bilibili
目录
1. 前言
大家好,这里是宋大水,今天给大家分享的是人眼瞳孔检测模型,主要用到的技术包括YOLO目标检测模型、MySQL数据库和PyQt的检测界面。工作量有登录、注册、图片检测、视频检测和摄像头流的检测,五次对比试验(YOLOv5s,YOLOv8s,YOLO11s,YOLO12s和基于transformer的RT-DETR)。
2. 项目展示
2.1 五次对比试验
仅展示部分数据,文档中包含所有数据
| P | R | mAP | mAP50-95 | 参数量(M) | GFLOPs | FPS(4060TI) | |
| YOLOv5s | 1 | 1 | 0.995 | 17.6 | 23.8 | 137.49 | |
| YOLOv8s | 1 | 1 | 0.995 | 0.85 | 21.4 | 28.4 | 140.05 |
| YOLO11s | 1 | 0.995 | 0.855 | 18.2 | 139.50 | ||
| YOLO12s | 1 | 0.994 | 0.995 | 0.851 | 21.2 | 116.70 | |
| RT-DETR | 0.999 | 1 | 0.841 | 63.1 | 103.5 | 57.06 |
选择了目前主流的v5,v8,11,12和RT-DERT这五个模型进行训练,从表中可以看出RT-DERT它的FPS的数值很低,所以如果想要一个在精度和推理速度相平衡的模型来说,YOLO系列检测模型无疑还是比较好的选择,在YOLO系列的这四个模型中,YOLOv8s的检测性能相对来说是最好的。
2.2 登录界面

这个界面由PyQt设计,具备两个功能,一个是登录功能,一个是注册功能,使用了MySQL数据库,因此我们在登录前首先要配置好数据库,才可以进行注册和登录,登录进去后进行目标检测功能。
2.3 检测界面

这个检测界面也是由PyQt所设计的,所以可以选择四个检测模型pt文件,分别是v5,v8,11和12,选择需要检测的资源,例如图片,视频,摄像头和rtsp流,设置好置信度和IOU,选择是否要将检测结果保存到文件夹,然后点击下方的推理按钮即可开始推理检测。
2.4 训练结果部分展示
在这里展示一些重要的训练结果图片(F1曲线、P-R曲线、训练结果和训练图片)





2.5 资源获取
1.b站搜宋大水(可从置顶视频链接直接跳转),视频置顶链接点击购买 49.9元。
2.直接点击下方购买链接:
3. 代码运行
3.1 数据集
本实验数据集是一个人眼瞳孔数据集,1个类别,部分图像如下图所示。
data.yaml配置文件如下:
3.2 python环境配置
本代码运行需要安装一些深度学习包,例如PyTorch,Torchvision,opencv,ultralytics等。
深度学习环境的三种搭建方式(彻底理解安装逻辑和步骤)
大家可以看上面这个视频学习如何安装深度学习环境。
3.3 mysql数据库
因为登录和注册功能使用到了MySQL数据库,所以我们需要安装配置下这个数据库,只需用该文件中的安装包安装mysql8数据库,然后将下面这个sql语句导入到数据库中,并且配置下LoginWindows.py文件的数据库用户名和密码即可使用。
这就是本次给大家分享的目标检测项目,如需更加详细的信息,可以浏览指定的视频。
更多推荐
所有评论(0)