ultralytics+YOLO实现马路高速公路车流量出入方向计数器实时统计示例代码

代码语言:python

所属分类:其他

代码描述:ultralytics+YOLO实现马路高速公路车流量出入方向计数器实时统计示例代码,支持80种不同物体的计数跟踪统计。

代码标签: ultralytics YOLO 马路 高速 公路 车流量 出入 方向 计数器 实时 统计 示例代码

下面为部分代码预览,完整代码请点击下载或在bfwstudio webide中打开

import cv2
import numpy as np
from ultralytics import YOLO

# ---------------- 配置参数 ----------------
# 视频路径 (请替换为你自己的视频路径,或者填 0 使用摄像头)
VIDEO_SOURCE = "traffic_video.mp4" 

# 模型选择 (yolov8n.pt 速度快,yolov8m.pt 精度高)
MODEL_NAME = "yolov8n.pt"

# 检测目标类别 ID (COCO数据集: 0=人, 2=汽车, 3=摩托车, 5=巴士, 7=卡车, 47=苹果)
# 如果你想检测汽车,使用 [2, 5, 7];如果想检测人,使用 [0]
TARGET_CLASSES = [2, 5, 7] 

# 计数线的设置 (这是一个水平线的例子,格式: [x1, y1, x2, y2])
# 注意:你需要根据视频的分辨率调整这条线的位置
LINE_Y = 500  # 线在Y轴的高度
LINE_START_X = 0
LINE_END_X = 1280 # 假设视频.........完整代码请登录后点击上方下载按钮下载查看

网友评论0