三帧差分法Python代码

以下是一个简单的三帧差分法的Python代码示例：

import cv2

# 读取第一帧
cap = cv2.VideoCapture('video.mp4')
ret, frame1 = cap.read()

# 读取第二帧
ret, frame2 = cap.read()

while cap.isOpened():
    # 读取第三帧
    ret, frame3 = cap.read()

    # 如果没有读取到第三帧，结束循环
    if not ret:
        break

    # 将三帧转换为灰度图像
    gray1 = cv2.cvtColor(frame1, cv2.COLOR_BGR2GRAY)
    gray2 = cv2.cvtColor(frame2, cv2.COLOR_BGR2GRAY)
    gray3 = cv2.cvtColor(frame3, cv2.COLOR_BGR2GRAY)

    # 计算第一帧和第二帧的差异
    diff1 = cv2.absdiff(gray1, gray2)

    # 计算第二帧和第三帧的差异
    diff2 = cv2.absdiff(gray2, gray3)

    # 计算两个差异图像的加权平均值
    diff = cv2.addWeighted(diff1, 0.5, diff2, 0.5, 0)

    # 对差异图像进行二值化处理
    thresh = cv2.threshold(diff, 25, 255, cv2.THRESH_BINARY)[1]

    # 对二值化图像进行腐蚀和膨胀处理
    thresh = cv2.erode(thresh, None, iterations=2)
    thresh = cv2.dilate(thresh, None, iterations=2)

    # 显示结果
    cv2.imshow('frame', frame3)
    cv2.imshow('diff', diff)
    cv2.imshow('thresh', thresh)

    # 更新帧
    frame1 = frame2
    frame2 = frame3

    if cv2.waitKey(25) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()

以上代码通过读取视频文件，逐帧计算差异，并对差异进行二值化和腐蚀膨胀处理，最终显示出原始帧、差异图像和阈值化图像三个窗口。该代码只是一个简单的示例，实际应用中可能需要根据具体情况进行一些优化和改进。