import numpy as np
import cv2
#经典的测试视频
cap = cv2.VideoCapture('vtest.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)
#寻找视频中的轮廓
fgmask_gray,contours,hierarchy=cv2.findContours(fgmask,cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)
for i in contours:
#计算各个轮廓的长度
perimeter = cv2.arcLength(i,True)
if perimeter > 188 :
#找到一个直矩形(不会旋转)
x,y,w,h = cv2.boundingRect(i)
cv2.rectangle(frame,(x,y),(x+w,y+h),(0,255,0),3)
cv2.imshow('frame',frame)
cv2.imshow('fgmask',fgmask)
k=cv2.waitKey(100)&0xff
if k == 27:
break
cap.release()
cv2.destroyAllWindows()
OpenCV-Python项目实战:Background Subtraction
最新推荐文章于 2023-11-24 14:05:11 发布