混合高斯模型
识别视屏流中运动的人代码实现
import numpy as np
import cv2
cap=cv2.VideoCapture(‘test.avi’) # 视频地址
#形态学操作需要使用(弄出来椭圆的结构)
kernel=cv2.getStructuringElement(cv2.MORPH_ELLIPSE,(3,3)) #MORPH_ELLIPSE为椭圆
(getStructuringElement函数—获取结构化元素)
#创建混合高斯模型用于背景建模
fgbg=cv2.createBackgroundSubtractorMOG2()
while(True):
ret, frame=cap.read()#读取图像为帧
fgmask=fgbg.apply(frame)#应用分离背景模型对图片处理
#形态学开运算去噪点
fgmask=cv2.morphologyEx(fgmask,cv2.MORPH_OPEN,kernel)
#寻找视屏中的轮廓
im,contours,hierarchy=cv2.findContours(fgmask,cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)
#因为行人有差不多的身高,所以用周长来确定行人
for c in contours:
#j计算轮廓的周长
perimeter=cv2.arcLength(c,True)
if perimeter > 190
#找到一个直矩形
x,y,w,h=cv2.boundingRect©
#画出这个矩形
cv2.rectangle(frame,(x,y),(x+w,y+h),(0,0,255),2)
cv2.imshow(‘frame’,frame)
cv2.imshow(‘fgmask’,fgmask)
k=cv2.waitKey(100) &0xff
if k==27:
break
cap.release()
cv2.destroyAllWindows()