OpenCV3使用
配置如下:
- opencv3.4.0
- imutils 0.4.5
简单的运动追踪
import argparse
import datetime
import imutils
import time
import cv2
ap = argparse.ArgumentParser()
ap.add_argument("-v","--video",help="path to the video file")
ap.add_argument("-a","--min-area",type=int,default=500,help="minimul area size")
args = vars(ap.parse_args())
if args.get("video",None) is None:
camera = cv2.VideoCapture(0)
time.sleep(0.25)
else:
camera = cv2.VideoCapture(args["video"])
firstFrame = None
while True:
(grabbed,frame) = camera.read()
if not grabbed:
break
frame = imutils.resize(frame,width=600)
gray = cv2.cvtColor(frame,cv2.COLOR_BGR2GRAY)
gray = cv2.GaussianBlur(gray,(21,21),0)
if firstFrame is None:
firstFrame = gray
continue
frameDelta = cv2.absdiff(firstFrame,gray)
thresh = cv2.threshold(frameDelta,25,255,cv2.THRESH_BINARY)[1]
thresh = cv2.dilate(thresh,None,iterations = 2)
(img,cnts, hierarchy) = cv2.findContours(thresh.copy(), cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)
for c in cnts:
if cv2.contourArea(c)<args["min_area"]:
(x,y,w,h) = cv2.boundingRect(c)
(x,y,w,h) = cv2.boundingRect(c)
cv2.rectangle(frame,(x,y),(x+w,y+h),(0,255,0),2)
text = "Occipied"
cv2.putText(frame,"Room Status:{}".format(text),(10,20),cv2.FONT_HERSHEY_SIMPLEX,0.35,(0,0,255),1)
cv2.putText(frame,datetime.datetime.now().strftime("%A %d %B &y %I:%M:%s%s"),(10,frame.shape[0]-10),cv2.FONT_HERSHEY_SIMPLEX,0.35,(0,0,255),1)
cv2.imshow("Security Feed",frame)
cv2.imshow("Thresh",thresh)
cv2.imshow("Frame Delta",frameDelta)
key = cv2.waitKey(1)&0xFF
if key == 27:
break
camera.release()
cv2.destroyAllWindows()
详情:https://www.pyimagesearch.com/2015/05/25/basic-motion-detection-and-tracking-with-python-and-opencv/