社区首页 >专栏 >有贼出没？看我用python来捉"贼"

有贼出没？看我用python来捉"贼"

赵云龙龙

发布于 2019-11-03 13:06:46

64000

代码可运行

文章被收录于专栏：python爱好部落python爱好部落

运行总次数：0

代码可运行

最近丢东西了，接二连三丢东西，很是糟心。是谁干的？真想把他给揪出来。自己用python捣鼓了个程序，看看效果。

最近自己的工位老是丢东西，关键只丢一样东西，而且每天早上来，桌子上都是乱七八糟的。像是遭了贼。

本来上班心情就不是那么愉快，这下心情更不爽了。也没丢啥值钱的东西，就是苹果手机充电的线。本来测试的时候也要用，就插在电脑上没拔，结果总是丢。

为这点破事跟人事说，也犯不着。也许谁拿去用用，说不定就还回来了。结果几个星期过去了，没有还的，只有丢的，接二连三，感觉是盯上我了。

丢到第四根的时候，扛不住了，害我自己的手机天天没地方充电。测试机也没法充电。

我得想办法把他找出来，找他要回来。我自己的电脑是一个macpro, 很大的屏，平时也不关机，它不有个摄像头么？我得从这上面下功夫。

import cv2

def make_photo():
    """使用opencv拍照"""
    cap = cv2.VideoCapture(0)  # 默认的摄像头
    while True:
        ret, frame = cap.read()
        if ret:
            cv2.imshow("capture", frame)  # 弹窗口
            # 等待按键q操作关闭摄像头
            if cv2.waitKey(1) & 0xFF == ord('q'):
                file_name = "theif.jpeg"
                cv2.imwrite(file_name, frame)
                break
        else:
            break

    cap.release()
    cv2.destroyAllWindows()


def make_video():
    """使用opencv录像"""
    cap = cv2.VideoCapture(0)  # 默认的摄像头
    # 指定视频代码
    fourcc = cv2.VideoWriter_fourcc(*"DIVX")
    out = cv2.VideoWriter('thief.mp4', fourcc, 20.0, (640, 480))
    while (cap.isOpened()):
        ret, frame = cap.read()
        if ret:
            out.write(frame)
            #
            cv2.imshow('frame', frame)
            # 等待按键q操作关闭摄像头
            if cv2.waitKey(1) & 0xFF == ord('q'):
                break
        else:
            break
    cap.release()
    out.release()
    cv2.destroyAllWindows()


if __name__ == '__main__':
    make_video()
    make_photo()

运行起来似乎还可以。

但是有个问题，如果我一直开机，一直拍照或者录制视频。感觉存储空间会抗不住。如果镜头有变化，才录制，这样会减少很多无用的录制，而且回放也简单很多。改进一下：

import cv2
import numpy, os, sys


def isPicChanged(dividePar, pointDelta, judgeTh):
    '''
    通过前后帧对比，判断画面是否改变
    :param dividePar = 40 # 对比间隔点，减少计算量
    :param pointDelta = 500 # 像素点的差异大于该值认为是差异点
    :param judgeTh = 640 # 判断变化画面大小的阈值：画面(1/judgeTh)
    '''

    capIdx = 0  # 截图命名
    camIdx = -1
    while (int(camIdx) < 0 or int(camIdx) > 10):
        print("enter camera index in 0 and 10:")
        camIdx = int(input())

    if not (os.path.isdir('cap')):  # 创建存放截图的文件夹
        os.system('mkdir -p {}'.format("cap"))

    cap = cv2.VideoCapture(camIdx)  # 调整参数实现读取视频或调用摄像头
    ret, frameBak = cap.read()
    for i in range(10):  # 刚打开相机时，曝光不稳定，清理10张
        ret, frameBak = cap.read()

    frameWidth = int(cap.get(cv2.WIDTH))
    frameHeight = int(cap.get(cv2.HEIGHT))
    print("frameWidth:{},frameHeight:{}".format(frameWidth, frameHeight))
    if frameWidth == 0:
        exit("camera is not available.")

    while True:
        frameBak = frame
        ret, frame = cap.read()
    cap.release()
    cv2.destroyAllWindows()

if __name__ == "__main__":
    isPicChanged(4, 50, 64)

测试一下，效果还真有。镜头变化了，才录制。后面想想，还有个问题，如果镜头变化很久，存储也会吃紧。于是又写个程序，隔段时间，来检测图片的变化，如果相似度太高的，可以删除一部分。 ssim对比：

from skimage.measure import compare_ssim
import cv2

class CompareImage():

    def compare_image(self, path_image1, path_image2):

        imageA = cv2.imread(path_image1)
        imageB = cv2.imread(path_image2)

        grayA = cv2.cvtColor(imageA, cv2.COLOR_BGR2GRAY)
        grayB = cv2.cvtColor(imageB, cv2.COLOR_BGR2GRAY)

        (score, diff) = compare_ssim(grayA, grayB, full=True)
        print("SSIM: {}".format(score))
        return score


compare_image = CompareImage()
compare_image.compare_image("1.jpg", "2.jpg")

直方图对比：

import cv2
import numpy as py
 
img = cv2.imread("../image-030.png")
img1 = cv2.imread("../image-002.jpg")
# 计算图img的直方图
H1 = cv2.calcHist([img], [1], None, [256],[0,256])
H1 = cv2.normalize(H1, H1, 0, 1, cv2.NORM_MINMAX, -1) # 对图片进行归一化处理
 
# 计算图img2的直方图
H2 = cv2.calcHist([img1], [1], None, [256],[0,256])
H2 = cv2.normalize(H2, H2, 0, 1, cv2.NORM_MINMAX, -1)
 
# 利用compareHist（）进行比较相似度
similarity = cv2.compareHist(H1, H2,0)
print(similarity)