软件界面如下所示:
功能:
模型选择
输入选择(本地文件、摄像头、RTSP视频流)
IoU调整
置信度调整
帧间延时调整
播放/暂停/结束
统计检测结果
详细介绍:
1.首先进行模型的选择(官网可下载),包含四种,分别是yolov5s.pt、yolov5m.pt、yolov5l.pt和yolov5x.pt。
2.选择置信度、IoU和帧间延时
3.系统支持输入图片、视频、摄像头和RTSP视频流的目标检测,其中,可以对图片进行处理,包括灰度化、平滑处理、均衡化、形态学、图像梯度、阈值处理、边缘检测、轮廓检测、直线检测、亮度调节和伽玛校正。
软件效果:
图片目标检测
摄像头目标检测
图像处理
说明:
本仓库的yolov5版本为v5.0,是直接从官方仓库拉取的,支持训练。
本仓库依赖模型有yolov5s.pt、yolov5m.pt、yolov5l.pt、yolov5x.pt,下载地址:https://github.com/ultralytics/yolov5/releases/tag/v5.0 点击地址后翻到最下面有下载链接,将下载好的模型放在pt文件夹下,运行界面时,会自动检测已有模型。
运行main.py开启检测界面后会自动检测已有模型。
问题汇总:
Q: 把模型替换为yolov5最新版本的模型后,界面左下角有错误提示,但是没有报错,请问怎么解决?
A: 使用v5.0之后的版本会报错”The size of tensor a (80) must match the size the size of tensor b(52) at non-singleton dimension 3″或者”AttributeError: Can’t get attribute ‘SPPF’ on “。模型版本不同则无法直接使用,建议用本仓库训练,或者修改代码以匹配你的模型。如果你想看详细错误报告,可以把DetThread类中的异常捕获取消,或者调用cgit模块(自行搜索cgit的用法,很简单)。
Q: 遇到错误:Permission denied:’pt’
A: 可能是pt文件在下载时有损坏,重新下载,或者是没有将pt文件放到pt文件夹中。
Q: ‘Upsample’ object has no attribute ‘recompute_scale_factor’
A: torch版本太高,下降一下版本吧,我的是torch=1.7.1、torchvision=0.8.2。torch==1.9.0、torchvision==0.10.0也可以。
Q: 如何打包成exe文件?
A: 打包过一次,但是启动后打不开,打包的文件1G左右,不建议打包。如果有人打包成功了,可以留个言。如果你只是不想使用命令行启动文件,windows用户可以建一个bat文件快速启动。
Q: 点击摄像头按钮后,再检测,为什么还是检测上一次的文件?
A: 点击摄像头按钮后,会自动检测电脑连接了几个摄像头(除了电脑自带摄像头,有些人还会连接usb摄像头),检测完成后,摄像头按钮下方会出现数字序号,你需要手动点击出现的数字选择摄像头。
Q: 请问怎么更改背景图片、背景颜色?
A: 自行搜索:pyqt5+qss。
如果您感兴趣,可以私信我,我会发您!祝大家学习进步!