import cv2
import numpy as np

# 创建背景建模器 (MOG2)
bg_subtractor = cv2.createBackgroundSubtractorMOG2(history=500, varThreshold=16, detectShadows=True)

cap = cv2.VideoCapture("your_video.mp4")  # 或者用图片序列

while True:
    ret, frame = cap.read()
    if not ret:
        break

    # 应用背景建模器
    fg_mask = bg_subtractor.apply(frame)

    # 获取背景图像
    bg_image = bg_subtractor.getBackgroundImage()

    # 显示
    cv2.imshow("Frame", frame)
    cv2.imshow("Foreground Mask", fg_mask)
    if bg_image is not None:
        cv2.imshow("Background Model", bg_image)

    if cv2.waitKey(30) & 0xFF == 27:
        break

cap.release()
cv2.destroyAllWindows()