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

好的，我可以为您提供一个简单的车辆计数程序，基于 OpenCV 库和 Python 语言编写。

首先，需要安装 OpenCV 库。请在终端或命令提示符中输入以下命令：

pip install opencv-python

然后，您需要准备两个视频文件，一个用于测试算法，另一个用于训练模型。视频文件应该是包含车辆的交通场景。

接下来，您可以使用以下Python代码来编写车辆计数程序：

import cv2

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

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

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

# 初始化变量
count = 0
car_count = 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
        
        # 绘制边界框
        cv2.rectangle(frame, (x, y), (x+w, y+h), (0, 0, 255), 2)
        
        # 计算车辆数量
        count += 1
        
    # 显示视频帧
    cv2.imshow("frame", frame)
    
    # 按下ESC键退出循环
    if cv2.waitKey(1) == 27:
        break
    
    # 打印车辆数量
    if count > car_count:
        car_count = count
        print("车辆数量：", car_count)

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

代码中使用了 BackgroundSubtractor 对象来提取前景区域，然后使用形态学操作来去除噪声。接下来，通过查找轮廓来计算车辆数量，并绘制每个车辆的边界框。最后，将计算出的车辆数量打印到控制台上。

请注意，这只是一个简单的车辆计数程序，可能需要进行更多的优化和改进，以适应不同的场景和要求。