[Python+OpenCV项目实战]——背景建模

向右看齐 2022-11-26 07:59 242阅读 0赞

### 文章目录 ###

*  一、代码
 *  二、效果（视频格式）

# 一、代码 #

import numpy as np
    import cv2
    
    #经典的测试视频
    cap = cv2.VideoCapture('test.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
        if k == 27:
            break
    
    cap.release()
    cv2.destroyAllWindows()

# 二、效果（视频格式） #

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dhMXR6eQ_size_16_color_FFFFFF_t_70]

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dhMXR6eQ_size_16_color_FFFFFF_t_70 1]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dhMXR6eQ_size_16_color_FFFFFF_t_70]: /images/20221124/20b61b32b87d457b827ba0f159ca15af.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dhMXR6eQ_size_16_color_FFFFFF_t_70 1]: /images/20221124/8c2bb4a1b5b046afb3f7b98a28b345d3.png