YOLOv5目标检测可视化界面
项目简介
本项目利用PyQT5搭建了一个YOLOv5目标检测的可视化界面，并将其打包成可执行的exe程序。该界面支持图像、视频和摄像头的目标检测，适配YOLOv5的各个版本模型。
功能特点
- 图像检测：支持对单张图片进行目标检测。
- 视频检测：支持对视频文件进行目标检测。
- 摄像头检测：支持实时摄像头输入的目标检测。
- 界面美化：使用qdarkstyle对界面进行了美化，提升用户体验。
使用说明
- 图像检测：点击“图片检测”按钮，选择要检测的图片文件，系统将自动进行目标检测并在界面上显示结果。
- 视频检测：点击“视频检测”按钮，选择要检测的视频文件，系统将自动进行目标检测并在界面上显示结果。
- 摄像头检测：点击“摄像头检测”按钮，系统将调用摄像头进行实时目标检测，并在界面上显示结果。
项目结构
- models：存放YOLOv5模型构建相关程序。
- utils：存放绘图、数据加载等相关工具。
- UI：存放软件图标和界面设计文件。
- result：存放预测之后的图片或视频结果。
- weights：存放模型权重文件，默认使用YOLOv5官方提供的yolov5s.pt。
依赖环境
- Python 3.x
- PyTorch
- OpenCV
- PyQT5
安装与运行
- 克隆本仓库到本地。
- 安装所需的依赖库：
pip install -r requirements.txt。 - 运行主程序：
python main.py。
贡献
欢迎大家提出改进建议和贡献代码，共同完善这个项目。
许可证
本项目遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。