Python+opencv realizes motion detection, tracking and taking pictures

Required modules: opencv, time, datetime, os, easygui (the ui interface is not required)

    Purpose: Python+opencv realizes motion detection and tracking and takes pictures and stores them

 
 
import cv2
import time
import datetime
import os
import easygui

def mkdir(path):
    folder = os.path.exists(path)

    if not folder:  # 判断是否存在文件夹如果不存在则创建为文件夹
        os.makedirs(path)  # makedirs 创建文件时如果路径不存在会创建这个路径
        print
        ("---  new folder...  ---")
        print
        ("---  OK  ---")

    else:
        print
        ("---  There is this folder!  ---")


file = "D:\\CCTVlook"#保存位置
mkdir(file)
print("文件储存于D:\\CCTVlook")
# 选取摄像头,0为笔记本内置的摄像头,1,2···为外接的摄像头
camera=cv2.VideoCapture(0)
title = easygui.msgbox(msg="将于5s后开始记录摄像头移动情况!""\n""请离开保证背景稳定""\n"
                           ,title="运动检测追踪拍照",ok_button="开始执行")
msg=easygui.msgbox(msg="移动物体保存于D:\\CCTVlook")
time.sleep(5)#延迟5s执行
background = None#初始化背景
def nothing(x):
    pass

cv2.namedWindow("fps")  #新建一个窗口
cv2.createTrackbar('level','fps',21,255,nothing)    #新建阈值滑动条
shot_idx = 0

while True:
    text = "No Target"
    flat=0
    # 滑动条赋值
    kerne = cv2.getTrackbarPos('level', 'fps')
    if kerne%2==0 :
        kerne=kerne+1#解决滑动条赋值到高斯滤波器是偶数异常抛出
    (grabbed, frame) = camera.read()
    # 对帧进行预处理,先转灰度图,再进行高斯滤波。
    # 用高斯滤波对图像处理,避免亮度、震动等参数微小变化影响效果
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    gray = cv2.GaussianBlur(gray, (kerne, kerne), 0)
    # 将第一帧设置为整个输入的背景
    if background is None:
        background = gray
        continue
    # 当前帧和第一帧的不同它可以把两幅图的差的绝对值输出到另一幅图上面来
    frameDelta = cv2.absdiff(background, gray)
    # 二值化
    thresh = cv2.threshold(frameDelta, 25, 255, cv2.THRESH_BINARY)[1]
    # 腐蚀膨胀
    thresh = cv2.dilate(thresh, None, iterations=2)
    #取轮廓
    cnts = cv2.findContours(thresh.copy(), cv2.RETR_EXTERNAL,
                                    cv2.CHAIN_APPROX_SIMPLE)[-2]
    # 遍历轮廓
    for c in cnts:
        if cv2.contourArea(c) < 1800: # 对于较小矩形区域,选择忽略
            continue
        flat=1#设置一个标签,当有运动的时候为1
        # 计算轮廓的边界框,在当前帧中画出该框
        (x, y, w, h) = cv2.boundingRect(c)
        cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 0, 255), 2)
        text = "Find Target! save as D:\CCTVlook"
        print("Find Target!")
    cv2.putText(frame, 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)
    #cv2.imshow("Frame Delta", frameDelta)

    cv2.imshow("fps", frame)
    #cv2.imshow("Thresh", thresh)

    key = cv2.waitKey(1) & 0xFF

    # 如果q键被按下,跳出循环
    ch = cv2.waitKey(1)
    if key == ord("q"):
        break
        
    if flat==1:#设置一个标签,当有运动的时候为1
        fn = 'D:\CCTVlook\shot%d.jpg' % (shot_idx)
        cv2.imwrite(fn, frame)
        shot_idx += 1
        continue
camera.release()
cv2.destroyAllWindows()
Execute the program, and then a CCTVlook folder will be generated in the root directory of the D drive, which will save photos of moving objects
Secretly showing his head was also discovered

 
 
 
 
 
 
 
 
Fatty behind


Guess you like

Origin blog.csdn.net/dgut_guangdian/article/details/79762409