24_背景建模

# 背景建模

# 1. 帧差法

# 2. 混合高斯模型


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

import numpy as np
import cv2

# 经典的测试视频
cap = cv2.VideoCapture('D:/pycharm/pycharm-cope/opencv/resource/videos/02_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)
    # 寻找视频中的轮廓
    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()

 结果展示