opencv 识别运动物体

import cv2
import numpy as np

cap = cv2.VideoCapture('video.mp4')
try:
    import cv2
    backSub = cv2.createBackgroundSubtractorMOG2()
except AttributeError:
    backSub = cv2.bgsegm.createBackgroundSubtractorMOG()

#形态学kernel
kernel = cv2.getStructuringElement(cv2.MORPH_RECT,(5,5))

while True:
    ret, frame = cap.read()
    if ret:
        #灰度
        cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)

        #去噪
        blur = cv2.GaussianBlur(frame, (3,3), 5)

        #去背景
        mask = backSub.apply(blur)

        #腐蚀
        erode = cv2.erode(mask,kernel)

        #膨胀
        dilate = cv2.dilate(erode,kernel,iterations = 3)

        close = cv2.morphologyEx(dilate,cv2.MORPH_CLOSE,kernel)
        close = cv2.morphologyEx(close, cv2.MORPH_CLOSE, kernel)

        contours,h = cv2.findContours(close,cv2.RETR_TREE,cv2.CHAIN_APPROX_SIMPLE)

        for (i, c) in enumerate(contours):
            (x,y,w,h) = cv2.boundingRect(c)
            cv2.rectangle(frame, (x,y),(x+w,y+h),(0,0,255),2)

        cv2.imshow('video',frame)
       # cv2.imshow('erode',close)
    key = cv2.waitKey(1)
    if key == 27:
        break


cap.release()
cv2.destroyAllWindows()
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值