用代码实现三帧差分法实现运动估计

好的，以下是使用 Python 实现三帧差分法的代码示例：

import cv2

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

# 获取第一帧图像
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(gray2, gray1)
    ret, thresh1 = cv2.threshold(diff1, 10, 255, cv2.THRESH_BINARY)

    diff2 = cv2.absdiff(gray1, frame1)
    ret, thresh2 = cv2.threshold(diff2, 10, 255, cv2.THRESH_BINARY)

    # 叠加两个差分图像得到最终结果
    final_diff = cv2.bitwise_and(thresh1, thresh2)

    # 进行运动估计并绘制矩形框
    contours, _ = cv2.findContours(final_diff, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
    for contour in contours:
        (x, y, w, h) = cv2.boundingRect(contour)
        cv2.rectangle(frame2, (x, y), (x + w, y + h), (0, 255, 0), 2)

    # 显示结果
    cv2.imshow('Video', frame2)

    # 更新帧和灰度图像
    gray1 = gray2.copy()
    frame1 = frame2.copy()

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

cap.release()
cv2.destroyAllWindows()

希望这段代码可以帮助你实现三帧差分法的运动估计！