【亲测免费】 YOLOv5 + PyQt5 项目使用教程

1. 项目目录结构及介绍

yolov5_pyqt5/
├── data/
│   ├── images/
│   └── labels/
├── models/
│   ├── yolov5s.pt
│   └── yolov5m.pt
├── utils/
│   ├── general.py
│   ├── torch_utils.py
│   └── ...
├── weights/
│   └── best.pt
├── config/
│   └── config.yaml
├── main.py
├── detect.py
├── train.py
├── requirements.txt
└── README.md

目录结构说明：

• data/: 存放训练和测试数据集，包括图像和标签。
• models/: 存放预训练的YOLOv5模型文件。
• utils/: 存放各种工具函数和辅助代码。
• weights/: 存放训练好的模型权重文件。
• config/: 存放项目的配置文件。
• main.py: 项目的启动文件，负责启动GUI界面。
• detect.py: 目标检测的主要逻辑代码。
• train.py: 训练模型的脚本。
• requirements.txt: 项目依赖的Python库列表。
• README.md: 项目的说明文档。

2. 项目启动文件介绍

main.py

main.py 是项目的启动文件，负责初始化并启动PyQt5的GUI界面。以下是该文件的主要功能：

import sys
from PyQt5.QtWidgets import QApplication
from gui.main_window import MainWindow

if __name__ == "__main__":
    app = QApplication(sys.argv)
    window = MainWindow()
    window.show()
    sys.exit(app.exec_())

主要功能：

• 初始化应用: 使用 QApplication 初始化PyQt5应用。
• 创建主窗口: 创建 MainWindow 对象，该对象负责管理整个GUI界面。
• 显示窗口: 调用 window.show() 显示主窗口。
• 运行应用: 调用 app.exec_() 进入应用的主循环，等待用户交互。

3. 项目配置文件介绍

config/config.yaml

config.yaml 是项目的配置文件，用于配置各种参数，如模型路径、数据集路径、训练参数等。以下是配置文件的示例内容：

model:
  weights: weights/best.pt
  conf_thres: 0.5
  iou_thres: 0.45

data:
  train: data/images/train
  val: data/images/val
  test: data/images/test

train:
  epochs: 300
  batch_size: 16
  img_size: 640

配置项说明：

• model: 配置模型相关的参数，如权重文件路径、置信度阈值、IoU阈值等。
• data: 配置数据集的路径，包括训练集、验证集和测试集。
• train: 配置训练相关的参数，如训练轮数、批量大小、图像尺寸等。

通过修改 config.yaml 文件，可以灵活调整项目的运行参数，满足不同的需求。

---

以上是基于 https://github.com/zstar1003/yolov5_pyqt5.git 项目的使用教程，涵盖了项目的目录结构、启动文件和配置文件的详细介绍。希望这份文档能帮助你快速上手并使用该项目。