Skip to content

添加YOLO车辆分类统计功能#5438

Merged
iiiostar merged 17 commits intoOpenHUTB:mainfrom
iiiostar:main
Apr 18, 2026
Merged

添加YOLO车辆分类统计功能#5438
iiiostar merged 17 commits intoOpenHUTB:mainfrom
iiiostar:main

Conversation

@iiiostar
Copy link
Copy Markdown
Contributor

修改概述:

添加YOLO车辆分类统计功能,支持按车型(car/motorbike/bus/truck)分类计数

修改的详细描述

  1. 修改 yolo_vehicle_counter_carla.py,添加分类计数器 class_counts 和 crossed_by_class
  2. 重构 draw_tracks_and_count 函数,使用 nonlocal 声明变量作用域
  3. 车辆穿过计数线时,自动按车型分类统计
  4. 视频处理完成后,终端输出各类别车辆统计结果

经过了什么样的测试?

  1. 操作系统: Windows (win32)
  2. Python版本: 3.11.9 (venv311虚拟环境)
  3. YOLO模型: yolo11n.pt
  4. 测试场景: 使用Carla仿真生成车辆视频,YOLO成功检测并分类计数

运行效果

动图、视频、截图等

  • 修改文件: src/yolo11n_vehicle_counter/scripts/yolo_vehicle_counter_carla.py
  • 运行结果: 视频左上角显示TOTAL计数,终端输出分类统计(car/motorbike/bus/truck)
  • 示例输出:

    总计数: XX 辆车

    car: XX
    motorbike: XX
    bus: XX
    truck: XX

48017941784d09ab95e5a5f15caf7a3e.mp4
a22f1749e159a22f84a3525af51e4b80

@iiiostar iiiostar merged commit 962fa3c into OpenHUTB:main Apr 18, 2026
1 check passed
3kyoo

This comment was marked as low quality.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants