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()
下面是识别效果