opencv+python检测入侵物体

opencv+python检测入侵物体

去年暑假在家的时候，参考别人的博客，编写了这个程序，本想着继续完善一下 ，却一直没时间做。偶然翻到了这个程序，决定放上来。
因为时间太久，我也没办法把当初参考的博客放上来（其实是因为懒得找），如果大家需要，我会放上链接。
环境：opencv2.4+python2.7
编辑器：spyder

# -*- coding: utf-8 -*-
"""
Spyder Editor

This is a temporary script file.
"""

import cv2
import time
import datetime

cap = cv2.VideoCapture(0)

avg = None
lastUploaded = datetime.datetime.now()
motionCounter = 0
time.sleep(10)
while(True):
    # 逐帧获取图像
    tiestamp = datetime.datetime.now()
    ret, frame = cap.read()
    text = "unoccupied"
    if not ret:
        break


    # 对每帧图像进行操作
    gray = cv2.resize(frame,width=500)#调整大小
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)#变成灰色图像
    gray = cv2.GaussianBlur(gray,(21,21),0)#高斯滤波
    if avg is None:
        avg = gray.copy().astype("float")
        continue
    cv2.accumulateWeighted(gray,avg,0.5)
    # 显示处理后的图像
    cv2.imshow('frame',gray)
    #计算当前帧与第一帧的区别
    frameDelta = cv2.absdiff(gray,cv2.convertScaleAbs(avg))
   # cv2.imshow('first2',frameDelta)
   #填充孔洞
    thresh = cv2.threshold(frameDelta, 45, 255, cv2.THRESH_BINARY)[1]
    thresh = cv2.dilate(thresh, None, iterations=2)
    cv2.imshow('thresh',thresh)
    #查找轮廓
    contours, hierarchy = cv2.findContours(thresh.copy(), cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)[-2:] 
  #  cv2.imshow('thresh2',thresh.copy())
    for c in contours:
        # if the contour is too small, ignore it
        if cv2.contourArea(c) < 500:
           continue

        # 计算轮廓的边界框，在当前帧中画出该框
        (x, y, w, h) = cv2.boundingRect(c)
        cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 2)
        cv2.imshow('found',frame)
        text = "Occupied"
        cv2.putText(frame, "Room Status: {}".format(text), (10, 20),
        cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 255), 2)
        cv2.putText(frame, datetime.datetime.now().strftime("%A %d %B %Y %I:%M:%S%p"),
        (10, frame.shape[0] - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.35, (0, 0, 255), 1)
    #if text == "Occupied":
      #  if (timestamp-lastUploaded).second>=2:
           # motionCounter+=1
            #if motionCounter>=23:


    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

# When everything done, release the capture
cap.release()
cv2.destroyAllWindows()

里面调用的是笔记本自带的摄像头来获取图像，如果想要采用别的方式，需要更改一下。如果效果要想好一点儿，可能还需要调整一下参数。程序已经测过，没有问题。