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()
相关推荐
火山引擎边缘云2 分钟前
开启报名!火山引擎 x PICO-全国大学生物联网设计竞赛赛题发布
人工智能·物联网·aigc
机器学习Zero6 分钟前
自然语言处理(9)—— 共现词矩阵及Python实现
人工智能·python·自然语言处理·nlp
Json_6 分钟前
睡前小故事数据集分享
人工智能·数据分析·openai
墨顿7 分钟前
模型推理的性能优化
人工智能·深度学习·性能优化·模型推理
萧门竹巷17 分钟前
看到的都惊了!AI 工具集
人工智能
大有数据可视化18 分钟前
智慧城市:如同为城市装上智能大脑,开启智慧生活
人工智能·生活·智慧城市
活力板蓝根23 分钟前
QQ音乐的 AI 化转型范本:一款音乐 App 如何用大模型重构听歌体验?
人工智能·音视频·娱乐·用户体验
闭月之泪舞37 分钟前
OpenCv高阶(六)——图像的透视变换
人工智能·opencv·计算机视觉
掘金安东尼38 分钟前
以一敌百:没有 Infra 团队,那就打造自己的技术雷达
人工智能·github
计算所陈老师38 分钟前
基于论文的大模型应用:基于SmartETL的arXiv论文数据接入与预处理(三)
大数据·人工智能·信息抽取