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