OpenCV运动物体检测

OpenCV运动物体检测

欢迎访问我的博客sakuraの绘梨衣
直接上识别代码，只要安装了OpenCV，可直接复现。注释很详细

import cv2

# 形态学操作需要使用
kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (3, 3))
# 创建混合高斯模型用于背景建模
fgbg = cv2.createBackgroundSubtractorMOG2()

# 运动物体识别
def motion_identify(frame):

    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)


if __name__ == "__main__":

    # 读入视频
    cap = cv2.VideoCapture(0)

    # 设置摄像头参数
    cap.set(3, 256)
    cap.set(4, 256)
    cap.set(5, 60)

    while True:

        # 循环读取每一帧
        flag, frame = cap.read()

        # 读取失败
        if not flag:
            print("Camera error!")
            break

        motion_identify(frame)

        # 则每10毫秒停顿一次，按下q键退出（OxFF为"q"的ASCII码）
        if cv2.waitKey(10) & 0xFF == ord('q'):
            break

    cap.release()
    cv2.destroyAllWindows()

下面是识别效果