实战博客指引:
经过前四步已经成功地搭建YOLOv5环境并完成了交通标志识别系统。但是由于YOLOv5没有图形化界面,因此在此使用YOLOv5整合PyQt5制作一个简易的可视化界面。
预览一下,最终得到的界面长这样:
1.安装PyQt5
输入以下命令安装PyQt5:
pip install PyQt5
pip install PyQt5-tools
2.启动PyQt5
打开控制台,输入如下命令,即可打开Qt工具:
designer.exe
如果报错的话可以查看是否缺少了什么东西,然后使用pip安装再重新启动。
启动成功的话将得到如下图所示的界面:
选择Main Window,然后点击创建,即可得到如下界面:
3.布局创建
之后我们将添加两个按钮以及一个文本框,用于选择图像还是视频以及显示区域。
在上图所示的左边可选择我们需要的Push Button,用于创建按钮。而label可以通过上面的filter过滤得到。然后修改按钮的名称,得到上图所示的结果。
然后选择这两个按钮,右键选择垂直布局:
再选择按钮和文本框,使用水平布局。
为了自适应布局,也就是当放大、缩小窗口大小时,仍保持界面元素的比例,可以在空白位置,点击鼠标右键,选择 布局 --> 水平布局
然后再修改尺寸比例
最后调整按钮的宽高,即可完成布局样式。
然后进行保存,得到untitled.ui文件。
4.转换为py文件
在该文件的路径下启动终端,运行下列代码将其转换为py文件:
pyuic5.bat -o main.py untitled.ui
最后将相应按钮与相关代码绑定,即可实现相关功能。
5.样例演示
图片检测
点击图片检测,选择相关图片,然后得到如下结果:
6.项目源码
项目源代码可联系博主。