清除在视频帧边界移动的对象中的白色像素(GMM算法)

import cv2
import numpy as np

def remove_white_pixels(video_path):
    cap = cv2.VideoCapture(video_path)
    fgbg = cv2.createBackgroundSubtractorMOG2(history=100, varThreshold=40, detectShadows=False)

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

        # 应用背景减除
        fgmask = fgbg.apply(frame)

        # 创建一个掩码，用于清除白色像素
        _, mask = cv2.threshold(fgmask, 250, 255, cv2.THRESH_BINARY)
        mask_inv = cv2.bitwise_not(mask)

        # 应用掩码到原始帧
        result = cv2.bitwise_and(frame, frame, mask=mask_inv)

        cv2.imshow('Frame', result)
        if cv2.waitKey(30) & 0xFF == ord('q'):
            break

    cap.release()
    cv2.destroyAllWindows()

# 使用示例
remove_white_pixels('path_to_your_video.mp4')