【YOLOv5实战5】基于YOLOv5的交通标志识别系统-YOLOv5整合PyQt5

实战博客指引:

  1. 实战环境搭建
  2. 自定义数据集
  3. 模型训练
  4. 模型测试与评估
  5. YOLOv5整合PyQt5

经过前四步已经成功地搭建YOLOv5环境并完成了交通标志识别系统。但是由于YOLOv5没有图形化界面,因此在此使用YOLOv5整合PyQt5制作一个简易的可视化界面。

预览一下,最终得到的界面长这样:
【YOLOv5实战5】基于YOLOv5的交通标志识别系统-YOLOv5整合PyQt5_第1张图片

1.安装PyQt5

输入以下命令安装PyQt5:

pip install PyQt5
pip install PyQt5-tools

2.启动PyQt5

打开控制台,输入如下命令,即可打开Qt工具:

designer.exe

【YOLOv5实战5】基于YOLOv5的交通标志识别系统-YOLOv5整合PyQt5_第2张图片
如果报错的话可以查看是否缺少了什么东西,然后使用pip安装再重新启动。
启动成功的话将得到如下图所示的界面:
【YOLOv5实战5】基于YOLOv5的交通标志识别系统-YOLOv5整合PyQt5_第3张图片
选择Main Window,然后点击创建,即可得到如下界面:
【YOLOv5实战5】基于YOLOv5的交通标志识别系统-YOLOv5整合PyQt5_第4张图片

3.布局创建

之后我们将添加两个按钮以及一个文本框,用于选择图像还是视频以及显示区域。
在上图所示的左边可选择我们需要的Push Button,用于创建按钮。而label可以通过上面的filter过滤得到。然后修改按钮的名称,得到上图所示的结果。

然后选择这两个按钮,右键选择垂直布局:
【YOLOv5实战5】基于YOLOv5的交通标志识别系统-YOLOv5整合PyQt5_第5张图片

再选择按钮和文本框,使用水平布局。
【YOLOv5实战5】基于YOLOv5的交通标志识别系统-YOLOv5整合PyQt5_第6张图片
为了自适应布局,也就是当放大、缩小窗口大小时,仍保持界面元素的比例,可以在空白位置,点击鼠标右键,选择 布局 --> 水平布局
【YOLOv5实战5】基于YOLOv5的交通标志识别系统-YOLOv5整合PyQt5_第7张图片

然后再修改尺寸比例
【YOLOv5实战5】基于YOLOv5的交通标志识别系统-YOLOv5整合PyQt5_第8张图片

最后调整按钮的宽高,即可完成布局样式。
【YOLOv5实战5】基于YOLOv5的交通标志识别系统-YOLOv5整合PyQt5_第9张图片
然后进行保存,得到untitled.ui文件。

4.转换为py文件

在该文件的路径下启动终端,运行下列代码将其转换为py文件:

pyuic5.bat -o main.py untitled.ui

【YOLOv5实战5】基于YOLOv5的交通标志识别系统-YOLOv5整合PyQt5_第10张图片
最后将相应按钮与相关代码绑定,即可实现相关功能。

5.样例演示

图片检测
点击图片检测,选择相关图片,然后得到如下结果:
【YOLOv5实战5】基于YOLOv5的交通标志识别系统-YOLOv5整合PyQt5_第11张图片

你可能感兴趣的:(深度学习,qt,python,深度学习,pytorch)