import cv2

import time

import datetime

import os

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)

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