基于边缘检测的物体计数python代码

以下是基于边缘检测的物体计数的Python代码示例：

import cv2

# 读取视频文件
cap = cv2.VideoCapture('video.mp4')

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

# 初始帧
ret, frame = cap.read()

# 定义计数器
count = 0

while ret:
    # 应用BackgroundSubtractor
    fgmask = fgbg.apply(frame)

    # 边缘检测
    edges = cv2.Canny(fgmask, 50, 190)

    # 查找轮廓
    contours, hierarchy = cv2.findContours(edges, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

    # 绘制轮廓
    cv2.drawContours(frame, contours, -1, (0, 0, 255), 2)

    # 计数器加1
    count += len(contours)

    # 显示计数
    cv2.putText(frame, 'Count: {}'.format(count), (10, 50), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 255), 2)

    # 显示视频帧
    cv2.imshow('frame', frame)

    # 按下q键退出
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

    # 读取下一帧
    ret, frame = cap.read()

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

这段代码首先读取视频文件，并创建了一个BackgroundSubtractor对象来提取前景，然后使用Canny边缘检测算法对前景进行处理，找到物体的轮廓，并绘制在视频帧上，最后计数器加1并显示计数。按下q键退出程序。