# Traffic-Sign-Recognition-PyQt5-YOLOv5-GUI
**Repository Path**: summry/Traffic-Sign-Recognition-PyQt5-YOLOv5-GUI
## Basic Information
- **Project Name**: Traffic-Sign-Recognition-PyQt5-YOLOv5-GUI
- **Description**: YOLOv5交通标识检测
- **Primary Language**: Python
- **License**: Not specified
- **Default Branch**: main
- **Homepage**: None
- **GVP Project**: No
## Statistics
- **Stars**: 3
- **Forks**: 0
- **Created**: 2025-03-09
- **Last Updated**: 2025-08-07
## Categories & Tags
**Categories**: Uncategorized
**Tags**: Yolo, Qt5, YOLOv5
## README
Road Sign Recognition Project Based on YOLOv5 (YOLOv5 GUI)
English |
简体中文
训练策略
This system is a road sign recognition application leveraging YOLOv5🚀 😊. It employs a MySQL database 💽, PyQt5 for the interface design 🎨, PyTorch deep learning framework⚡. Additionally, it incorporates CSS styles 🌈.
The system comprises five key modules:
1. System Login Module 🔑: Responsible for user authentication.
2. Initialization Parameter Module 📋: Provides settings for initializing YOLOv5 model parameters.
3. Sign Recognition Module 🔍: The core functionality responsible for recognizing road signs and updating the database with the results.
4. Database Module 💾: Consists of two sub-modules - basic database operations and data analysis.
5. Image Processing Module 🖼️: Handles the processing of individual images and associated data.
The entire system is designed to support various data input methods and model switching. Additionally, it offers image enhancement techniques such as mosaic and mixup 📈.
## Screenshots
* ### Sign Recognition Module
`The three checkboxes in the lower left corner are results save, start database entry, and model visual analysis.`
* ### Image Processing and Data Augmentation Module
`The right column is a batch image data enhancement with custom parameters (using the checked data increment method for all images in a folder with a certain probability)`
* ### Parameter Initialization Module
`Model basic parameters Select Configure`
* ### Database Module
* ### Data Analysis Module
* ### Login Interface
### Video Demo
[YOUTUBE DEMO: Road Sign Recognition System Based on YOLOV5](https://youtu.be/qoHaXvp_Gxk?si=xIAm1UXCLTjR8kUD)
[BiliBili Demo: Road Sign Recognition System Based on YOLOV5](https://www.bilibili.com/video/BV1Ck4y1Y7Bk/?spm_id_from=333.999.0.0&vd_source=40d9cda43378fbc89cd5184e09bf1272)
### Install Dependencies
To install the required dependencies, run:
```bash
pip install -r requirements.txt
```
---
## **Quick Start**
### 1. **Setting Up the Database**
To run the application, you need to set up your MySQL database. Follow these steps to prepare your database:
- **Automatic Database Creation (Optional)**:
- If you prefer an automated setup, a batch script is provided. Run the **`setup_database.bat`** script to create the database. This requires MySQL to be installed and configured on your system.
- **Manual Database Creation**:
- Alternatively, you can manually create the database in MySQL. Import and execute the **`data/regn_mysql.sql`** file in your MySQL environment to set up the necessary database and tables.
### 2. **Configuring Database Connection in Code**
After setting up the database, update the connection Settings in the code to change the authentication information for your local database (these four variables are at the beginning of the code, around line 59, as follows); P.S. These authentication messages are called twice in the code (around lines 111 and 1783)
```python
# Database connection settings as global variables
DB_HOST = 'localhost' # Database host
DB_USER = 'root' # Database user
DB_PASSWORD = '1234' # Database password
DB_NAME = 'traffic_sign_recognition' # Database name
```
### **Note on Cryptography Package**
If you encounter a **`RuntimeError: 'cryptography' package is required for sha256_password or caching_sha2_password auth methods`**,
This is because the database authentication has gone wrong and the database needs to be properly created and the password entered.
This error will also be reported if you do not have the mysql service started locally, so make sure your mysql service is started.
### 3. Run `main.py`.
### 4. Enter your account and password to log in
Here are the default login credentials:
| Username | Password |
|----------|----------|
| admin | 123456 |
| 1 | 2 |
Or modify the main function in main.py: remove the logon logic to enter the system directly without authentication.
---
## Project Structure
- `pt` folder: Contains the YOLOv5 model file `best.pt` for road sign recognition.
- `main_with` folder: Contains `login.py` for the login UI and `win.py` for the main UI.
- `dialog` folder: Contains the RTSP pop-up interface.
- `apprcc_rc.py`: The resource file for the project.
- `login_ji.py`: Implements the login logic for the UI.
- `data/run/run-exp52`: The YOLOv5 road sign recognition model trained for 300 epochs.
- `utils/tt100k_to_voc-main` folder: Tool for converting JSON annotations to YOLO format.
- `result`: Folder to save inference results.
- `run`: Folder to save training logs and outputs.
- Dataset: Download from [TT100k : Traffic-Sign Detection and Classification in the Wild](https://cg.cs.tsinghua.edu.cn/traffic-sign/).
- Database files: Located in the `data` folder, see `-regn_mysql.sql` for setup.
> Since this project was done while I was learning YOLOv5 (quite a while ago), the main logic is concentrated in the main.py file. In other words, I didn't modularize different functions, and I didn't have a clear division of module structure. Now I want to divide it into modules, but I'm too lazy, ha ha :smile:. If you're interested, you can modularize it so it's clearer.
## Acknowledgements
- For converting the TT100K dataset to VOC format and selecting more than 100 images and XMLs for each category, see this [CSDN blog post](https://blog.csdn.net/Hankerchen/article/details/120727299?spm=1001.2014.3001.5502).
- The PyQt5-YOLOv5 integration was inspired by this [GitHub repository](https://github.com/Javacr/PyQt5-YOLOv5).
## Contact Us
> WeChat：AIGCSD
>
> email: 2545197649@qq.com
>
> Official Accounts：AI进修生
## Star History
Track the GitHub star history of this project:
