基于YOLOv5的道路标志识别项目(yolov5界面GUI)

在这里插入图片描述

基于YOLOv5的道路标志识别项目(yolov5界面GUI)

English | 简体中文

这是一个关于yolov5的道路标志识别项目,使用Pyqt5开发界面,Yolov5训练模型,数据库Mysql,包含五个模块:初始化参数、标志识别、数据库、数据分析和图像处理。

项目地址https://github.com/Ai-trainee/Traffic-Sign-Recognition-PyQt5-YOLOv5-GUI

软件截图

  • 标志识别模块

    在这里插入图片描述

左下角三个勾选框分别是结果保存、启动数据库录入、以及模型可视化分析

  • 图像处理与数据增强模块

    在这里插入图片描述

    右侧栏是自定义参数的批量图像数据增强(按一定概率对一个文件夹所有图片使用勾选的数据增加方法)

  • 初始化参数模块

    在这里插入图片描述

    模型基本参数勾选配置

  • 数据库模块

    在这里插入图片描述

  • 数据分析模块

    在这里插入图片描述

  • 登录界面

    在这里插入图片描述

演示视频

基于YOLOV5的道路标志识别系统

安装依赖

pip install -r requirements.txt


快速开始

1. 设置数据库

为了运行应用程序,您需要设置您的 MySQL 数据库。按照以下步骤准备您的数据库:

  • 自动数据库创建(可选)
    • 如果您喜欢自动设置,我们提供了一个批处理脚本。运行 setup_database.bat 脚本来创建数据库。这需要在您的系统上安装并配置 MySQL。
  • 手动数据库创建
    • 或者,您可以在 MySQL 中手动创建数据库。在您的 MySQL 环境中导入并执行 data/regn_mysql.sql 文件,以设置必要的数据库和表。

2. 在代码中配置数据库连接

设置数据库之后,更新代码中的连接设置;这4个变量在代码的开头,请更改成你本地数据库的身份验证信息;这些身份验证信息再代码中有两处调用(大约第111行和第1783行)

# 数据库连接设置作为全局变量
DB_HOST = 'localhost'    # 数据库主机
DB_USER = 'root'         # 数据库用户
DB_PASSWORD = '1234'     # 数据库密码
DB_NAME = 'traffic_sign_recognition'  # 数据库名

关于数据库链接的注意事项

如果遇到 RuntimeError: 'cryptography' package is required for sha256_password or caching_sha2_password auth methods 错误,这是因为数据库身份验证出错了,需要正确地创建数据库并输入密码。

3. 运行 main.py

4. 输入您的账号和密码以登录。

以下是默认的登录凭据:

用户名密码
admin123456
12

或者修改main.py中的主函数:删除登陆逻辑,以直接进入系统而无需进行身份验证。


项目模块

  • pt文件夹:存放模型(best.pt是道路标志识别模型)
  • main_with文件夹:login.py(登陆ui)、win.py(主ui)
  • dialog文件夹:rtsp弹出界面
  • apprcc_rc.py:资源文件
  • login_ji.py:界面登陆逻辑文件
  • data/run/run-exp52:300轮训练后的道路标志识别模型
  • utils/tt100k_to_voc-main文件夹:json转yolo格式
  • result保存一些推理文件,run保存训练文件
  • 数据集:TT100k : Traffic-Sign Detection and Classification in the Wild
  • 数据库文件:data文件夹下-regn_mysql.sql

由于这个项目是在我学习YOLOv5时完成的(已经过了很长一段时间),因此主要的逻辑代码都集中在main.py文件中。换句话说,我没有将不同功能模块化,没有进行模块结构的清晰划分。主要是由于我太懒了,嘻嘻 😄。如果您有兴趣,可以将其模块化,这样它就会更加清晰。

致谢

Star History

在这里插入图片描述

  • 37
    点赞
  • 32
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
要搭建YOLOv5的可视化界面,您可以使用PyQt库来创建GUI应用程序。下面是一个简单的例子,演示了如何使用PyQt搭建YOLOv5的可视化界面。 首先,您需要安装必要的库和YOLOv5。假设您已经安装了YOLOv5并且在Python中导入了它。接下来,安装PyQt库: ``` pip install PyQt5 ``` 然后,您可以使用以下代码创建一个简单的GUI应用程序,该应用程序将YOLOv5应用于图像并显示结果: ``` python import sys from PyQt5.QtCore import Qt from PyQt5.QtGui import QPixmap from PyQt5.QtWidgets import QApplication, QMainWindow, QWidget, QLabel, QPushButton, QFileDialog from yolov5.detect import detect_image # 假设您已经导入了YOLOv5 class MainWindow(QMainWindow): def __init__(self): super().__init__() # 创建主窗口 self.setWindowTitle("YOLOv5可视化界面") self.setGeometry(100, 100, 800, 600) # 创建中心窗口 self.centralWidget = QWidget() self.setCentralWidget(self.centralWidget) # 创建标签和按钮 self.imageLabel = QLabel(self.centralWidget) self.imageLabel.setGeometry(50, 50, 500, 500) self.imageLabel.setAlignment(Qt.AlignCenter) self.openButton = QPushButton("打开图片", self.centralWidget) self.openButton.setGeometry(600, 50, 120, 40) self.openButton.clicked.connect(self.open_image) self.detectButton = QPushButton("检测", self.centralWidget) self.detectButton.setGeometry(600, 150, 120, 40) self.detectButton.clicked.connect(self.detect_image) # 初始化图像 self.image = None def open_image(self): # 打开文件对话框 options = QFileDialog.Options() options |= QFileDialog.DontUseNativeDialog fileName, _ = QFileDialog.getOpenFileName(self,"打开图片", "","Images (*.png *.xpm *.jpg *.bmp)", options=options) if fileName: # 加载图像并显示 self.image = QPixmap(fileName) self.imageLabel.setPixmap(self.image) self.imageLabel.setScaledContents(True) def detect_image(self): if self.image is not None: # 将图像换为numpy数组并应用YOLOv5 result = detect_image(self.image.toImage().constBits().asarray(self.image.width() * self.image.height() * self.image.depth() // 8)) # 显示结果 print(result) if __name__ == '__main__': app = QApplication(sys.argv) window = MainWindow() window.show() sys.exit(app.exec_()) ``` 这个例子创建了一个简单的GUI应用程序,其中包含一个标签和两个按钮。单击“打开图片”按钮将打开文件对话框,允许您选择要处理的图像。单击“检测”按钮将应用YOLOv5,检测图像中的物体,并将结果打印到控制台中。 请注意,这只是一个简单的例子,您需要根据您的具体需求进行修改。例如,您可能需要添加更多的按钮和控件来控制YOLOv5的参数,或者将结果显示在GUI中而不是打印到控制台中。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Aitrainee

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值