友情提示:官网的yolov5版本会持续更新,比如最近新增了SPPF类。为了避免不兼容的问题,建议使用本仓库的yolov5。如果想兼容最新版本的yolov5,自行更改对应的代码即可,改动不大。另外,最新的界面添加了背景图片,背景图片的存在会影响界面的流畅度,如果使用者发现界面使用起来不流畅,可以将yolo_winpy中的148行paintEvent函数注释掉。
本仓库的yolov5版本为v5.0,由于是直接从yolov5仓库下拉取下来的源码,本仓库也支持训练。模型下载地址:https://github.com/ultralytics/yolov5/releases/tag/v5.0 翻到最下面有链接下载. 2021/10/8: 所有代码已上传,直接clone后,运行yolo_win.py即可开启界面。
界面是在ultralytics的yolov5基础上建立的,界面使用pyqt5实现,内容较简单,娱乐而已。
功能:
- 模型选择
- 本地文件选择(视频图片均可)
- 开关摄像头
- 运行/终止
- 统计检测结果
默认模型为yolov5s.pt,默认输入文件为电脑摄像头视频
使用视频: https://www.bilibili.com/video/BV1sQ4y1C7Vk?spm_id_from=333.999.0.0
csdn: https://blog.csdn.net/weixin_41735859/article/details/120507779?spm=1001.2014.3001.5501
摄像头检测画面:
运行yolo_win.py即可开启检测界面。
存在的一个小问题,切换模型或者文件过于频繁,可能会卡住,重启一下即可。
这种情况很少出现,问题不大。