import numpy as np
import cv2

# 经典的测试视频
cap = cv2.VideoCapture('02_Video/01_Foreground.avi')

# 形态学操作需要使用
kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE,(3,3))
# 创建混合高斯模型用于背景建模
fgbg = cv2.createBackgroundSubtractorMOG2() # 混合高斯模型实例化对象
while(True):    
    ret,frame = cap.read()    
    fgmask = fgbg.apply(frame)    # 每一帧应用到混合高斯模型中
    # 形态学开运算去噪点
    fgmask = cv2.morphologyEx(fgmask,cv2.MORPH_OPEN,kernel)
    # 寻找视频中的轮廓
    im, contours, hierarchy = cv2.findContours(fgmask, cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)    
    
    for c in contours:
        # 计算各轮廓的周长
        perimeter = cv2.arcLength(c,True)
        
        if perimeter > 188:
            # 找到一个直矩形 (不会旋转)
            x, y, w, h = cv2.boundingRect(c)
            # 画出这个矩形
            cv2.rectangle(frame,(x,y),(x+w,y+h),(0,255,0),2)
            
    cv2.imshow('frame',frame)
    cv2.imshow('fgmask',fgmask)
    k = cv2.waitKey(150) & 0xff # 0xff 表示按退出键 ESC 就停止了
    if k == 27:
        break

cap.release()
cv2.destroyAllWindows()
背景建模¶

1. 帧差法¶

2. 混合高斯模型¶

2.1 混合高斯模型学习方法¶

2.2 混合高斯模型测试方法¶
