三帧差分法Python代码

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

import cv2

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

# 定义帧数和阈值
frame_count = 0
threshold = 30

# 读取第一帧
ret, frame1 = cap.read()

# 转换为灰度图像
gray1 = cv2.cvtColor(frame1, cv2.COLOR_BGR2GRAY)

# 循环读取视频帧
while True:
    # 读取下一帧
    ret, frame2 = cap.read()
    
    # 如果视频结束，退出循环
    if not ret:
        break
    
    # 转换为灰度图像
    gray2 = cv2.cvtColor(frame2, cv2.COLOR_BGR2GRAY)
    
    # 计算第一帧和第二帧的差异
    diff1 = cv2.absdiff(gray1, gray2)
    
    # 读取第三帧
    ret, frame3 = cap.read()
    
    # 如果视频结束，退出循环
    if not ret:
        break
    
    # 转换为灰度图像
    gray3 = cv2.cvtColor(frame3, cv2.COLOR_BGR2GRAY)
    
    # 计算第二帧和第三帧的差异
    diff2 = cv2.absdiff(gray2, gray3)
    
    # 计算前两帧的差异和当前帧的差异
    diff = cv2.bitwise_and(diff1, diff2)
    
    # 应用阈值，只保留像素值大于阈值的像素
    thresholded = cv2.threshold(diff, threshold, 255, cv2.THRESH_BINARY)[1]
    
    # 显示结果
    cv2.imshow('frame', thresholded)
    
    # 更新前两帧
    gray1 = gray2
    gray2 = gray3
    
    # 按下q键退出循环
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

# 释放视频捕获器
cap.release()

# 关闭窗口
cv2.destroyAllWindows()

这个代码使用OpenCV库读取视频文件，然后对每个帧执行以下步骤：

1. 将帧转换为灰度图像。
2. 计算前两帧和当前帧的差异。
3. 应用阈值，只保留像素值大于阈值的像素。
4. 显示结果。

这个代码还包括一个while循环，它会一直读取视频帧，直到视频结束或按下q键。