车流量统计系统

一个基于 YOLO 和 OpenCV 的智能车流量统计系统，支持多条检测线的车辆计数和分类统计。

功能特点

• 支持多种车辆类型检测（汽车、摩托车、公交车、卡车）
• 支持多条检测线设置
• 实时车辆计数和分类统计
• 高精度的车辆轨迹追踪
• 详细的统计报告生成
• 可视化检测结果

安装要求

• Python 3.7+
• OpenCV
• Ultralytics YOLO
• NumPy

快速开始

# 克隆项目
git clone https://github.com/yourusername/traffic-flow-counter.git
cd traffic-flow-counter

# 安装依赖
pip install -r requirements.txt

# 运行程序
python main.py

使用方法

1. 运行程序后会打开设置界面
2. 在视频第一帧上点击两点设置检测线
3. 可设置多条检测线，每条线用不同颜色标识
4. 按 Enter 完成设置，开始车流量统计

项目结构

traffic-flow-counter/
 src/                    # 源代码
    line_drawer.py     # 检测线绘制模块
    vehicle_tracker.py # 车辆追踪模块
    counter.py         # 计数模块
    utils.py           # 工具函数
 config/                # 配置文件
 data/                  # 输入数据
 output/                # 输出结果
 docs/                  # 文档
 main.py               # 主程序
 requirements.txt      # 依赖包列表
 README.md            # 项目说明

许可证

MIT License