用Python编写一个分车道车辆计数程序

好的，以下是一个用Python编写的分车道车辆计数程序，需要使用OpenCV库和numpy库。

在代码中，我们需要先对车道进行分割，然后对每个车道分别进行车辆计数。具体实现如下：

import cv2
import numpy as np

# 设置输入视频文件
video_file = "traffic_scene.mp4"

# 创建 VideoCapture 对象
cap = cv2.VideoCapture(video_file)

# 创建 BackgroundSubtractor 对象
fgbg = cv2.createBackgroundSubtractorMOG2()

# 定义车道线的位置
lane1_start = 150
lane1_end = 300
lane2_start = 350
lane2_end = 500

# 初始化变量
count_lane1 = 0
count_lane2 = 0

# 循环处理每一帧视频
while True:
    # 读取一帧视频
    ret, frame = cap.read()
    
    # 如果视频结束，则退出循环
    if not ret:
        break
    
    # 转换为灰度图像
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    
    # 对灰度图像进行高斯模糊
    blur = cv2.GaussianBlur(gray, (21, 21), 0)
    
    # 背景减除
    fgmask = fgbg.apply(blur)
    
    # 进行形态学操作
    kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (5, 5))
    closing = cv2.morphologyEx(fgmask, cv2.MORPH_CLOSE, kernel)
    opening = cv2.morphologyEx(closing, cv2.MORPH_OPEN, kernel)
    
    # 查找轮廓
    contours, hierarchy = cv2.findContours(opening, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
    
    # 绘制轮廓
    cv2.drawContours(frame, contours, -1, (0, 255, 0), 2)
    
    # 分别计算车道数量
    for cnt in contours:
        # 计算轮廓的边界框
        x, y, w, h = cv2.boundingRect(cnt)
        
        # 忽略太小的边界框
        if w < 50 or h < 50:
            continue
        
        # 判断车辆所在的车道
        if x > lane1_start and x + w < lane1_end:
            count_lane1 += 1
        elif x > lane2_start and x + w < lane2_end:
            count_lane2 += 1
        
        # 绘制边界框
        cv2.rectangle(frame, (x, y), (x+w, y+h), (0, 0, 255), 2)
    
    # 绘制车道线
    cv2.line(frame, (lane1_end, 0), (lane1_end, frame.shape[0]), (255, 0, 0), 2)
    cv2.line(frame, (lane2_start, 0), (lane2_start, frame.shape[0]), (255, 0, 0), 2)
    
    # 显示视频帧
    cv2.imshow("frame", frame)
    
    # 按下ESC键退出循环
    if cv2.waitKey(1) == 27:
        break
    
    # 打印车辆数量
    print("车道1车辆数量：", count_lane1)
    print("车道2车辆数量：", count_lane2)

# 释放资源
cap.release()
cv2.destroyAllWindows()

在代码中，我们首先定义了两个车道的位置。然后，对于每个轮廓，判断车辆是否在车道内，如果在车道内，则对相应的车道计数。最后，绘制车道线和边界框，并将车道数量打印到控制台上。

需要注意的是，该程序可能需要根据实际情况进行调整和优化，以达到更好的车辆计数效果。