Skip to content

基于 YOLOv5 与 PyQt5 的智能车辆感知与采集框架。集成实时目标检测(人/车)与 GNSS/IMU 数据融合,提供可视化仪表盘与数据记录功能。

License

Notifications You must be signed in to change notification settings

aprlost/YOLO-VPP

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

18 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

基于 YOLO 的智能车在途感知框架


Python PyTorch YOLOv5 PyQt5 License

一个集成了CV与传感器融合的智能车载感知框架——利用 YOLOv5 进行实时目标检测,结合 GNSS/IMU 串口通信实现车辆状态的全方位感知。


主要工作内容:标注并训练人车数据集,基于 YOLO 实现视频流目标的识别与计数;开发 Qt 界面与串口程序解析 GNSS/IMU 数据,实时解算并显示人车定位状态。 本项目旨在解决智能交通系统中车辆对周界环境感知的不确定性问题。系统通过软硬件协同,实现了以下核心能力:

  1. 视觉感知:利用 YOLOv5 深度学习算法,对车载摄像头视频流中的“行人”和“车辆”进行实时识别与计数。
  2. 多源信息融合:通过串口通信解析 NMEA-0183 协议,实时获取并显示车辆的经纬度、速度、海拔、航向及卫星状态。
  3. 交互式终端:基于 PyQt5 设计了可视化 GUI,支持视频录制、截图保存以及各类传感器数据的实时仪表盘显示。
  • 实时目标检测
    • 集成 YOLOv5 模型,支持 Webcam 或视频流输入。
    • 实时统计画面中的 车辆 (Car)行人 (Person) 数量。
  • GNSS 数据解析
    • 支持 $GPGGA$GPRMC 语句解析。
    • 实时显示:经纬度、地面速度、UTC时间转北京时间、卫星数量。
  • 现代化 GUI 界面
    • 实时视频回传显示。
    • 仪表盘数据显示。
    • 包含 截图录像 功能。
  • 数据记录
    • 支持检测结果的视频保存。
    • 支持关键帧截图保存。

📸 效果展示 (Demo)

场景1
场景 1
场景2
场景 2
场景3
场景 3
场景4
场景 4

1. 克隆仓库

git clone https://github.com/aprlost/YOLO-VPP.git
cd YOLO-VPP

2. 环境配置

建议使用 Conda 创建独立环境以避免依赖冲突:

conda create -n vehicle-perception python=3.8
conda activate vehicle-perception

3. 安装依赖

本项目依赖 YOLOv5 的基础环境以及 PyQt5 和 串口通信库。请确保安装以下核心库,以及代码内其他相关库等:

pip install PyQt5 pyserial pynmea2 pyautogui opencv-python torch torchvision

4. 硬件连接

摄像头:确保 USB 摄像头已连接。

GNSS 模块:连接电脑。

注意:默认串口号配置为 COM16。如果你的设备端口不同,请务必打开 final.py 修改以下代码:

ser = serial.Serial(port="COM16", baudrate=115200, ...)

5. 运行系统

直接运行主程序即可启动 Qt 界面:

python final.py

6. 操作指南

显示视频:点击界面上的“显示视频”按钮,加载 YOLO 模型并开始推理。

显示文本:点击“显示文本”按钮,开始读取并解析串口 GPS 数据。

功能按钮:

截图:保存当前界面截图到本地。 录像:点击开始录制,再次点击停止录制(保存在项目根目录)。

📈 Star History

Star History Chart

About

基于 YOLOv5 与 PyQt5 的智能车辆感知与采集框架。集成实时目标检测(人/车)与 GNSS/IMU 数据融合,提供可视化仪表盘与数据记录功能。

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages