本文主要实现了伯乐在线上的一个实践小项目,原文链接,用以巩固opencv视频操作知识内容。整个项目均有代码注释,通俗易懂,短短几十行就可以达到还算不错的实现效果,做起来成就感满满哒。打开编辑器,一起来感受下opencv+python在CV中的无穷魅力吧 ^_^

 1 import argparse
 2 import time
 3 import imutils
 4 import cv2
 5 
 6 # 创建参数解析器并解析参数
 7 ap = argparse.ArgumentParser()
 8 ap.add_argument("-v", "--video", help="path of the video")
 9 # 待检测目标的最小面积,该值需根据实际应用情况进行调整(原文为500)
10 ap.add_argument("-a", "--min-area", type=int, default=2000, help="minimum area size")
11 args = vars(ap.parse_args())    #@
12 
13 # 如果video参数为空,则从自带摄像头获取数据
14 if args.get("video") == None:
15     camera = cv2.VideoCapture(0)
16 # 否则读取指定的视频
17 else:
18     camera = cv2.VideoCapture(args["video"])
19 
20 
21 # 开始之前先暂停一下,以便跑路(离开本本摄像头拍摄区域^_^)
22 print("Ready?")
23 time.sleep(1)
24 print("Action!")
25 
26 # 初始化视频第一帧
27 firstRet, firstFrame = camera.read()
28 if not firstRet:
29     print("Load video error!")
30     exit(0)
31 
32 # 对第一帧进行预处理
33 firstFrame = imutils.resize(firstFrame, width=500)  # 尺寸缩放,width=500
34 gray_firstFrame = cv2.cvtColor(firstFrame, cv2.COLOR_BGR2GRAY) # 灰度化
35 firstFrame = cv2.GaussianBlur(gray_firstFrame, (21, 21), 0) #高斯模糊,用于去噪
36 
37 # 遍历视频的每一帧
38 while True:
39     (ret, frame) = camera.read()
40 
41     # 如果没有获取到数据,则结束循环
42     if not ret:
43         break
44 
45     # 对获取到的数据进行预处理
46     frame = imutils.resize(frame, width=500)
47     gray_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
48     gray_frame = cv2.GaussianBlur(gray_frame, (21, 21), 0)
49 
50     # cv2.imshow('video', firstFrame)
51     # 计算第一帧和其他帧的差别
52     frameDiff = cv2.absdiff(firstFrame, gray_frame)
53     # 忽略较小的差别
54     retVal, thresh = cv2.threshold(frameDiff, 25, 255, cv2.THRESH_BINARY)
55 
56     # 对阈值图像进行填充补洞
57     thresh = cv2.dilate(thresh, None, iterations=2)
58     image, contours, hierarchy = cv2.findContours(thresh.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
59 
60     text = "Unoccupied"
61     # 遍历轮廓
62     for contour in contours:
63         # if contour is too small, just ignore it
64         if cv2.contourArea(contour) < args["min_area"]:
65             continue
66 
67         # 计算最小外接矩形(非旋转)
68         (x, y, w, h) = cv2.boundingRect(contour)
69         cv2.rectangle(frame, (x, y), (x+w, y+h), (0,255,0), 2)
70         text = "Occupied!"
71 
72     cv2.putText(frame, "Room Status: {}".format(text), (10,20), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0,0,255), 2)
73 
74     cv2.imshow('frame', frame)
75     cv2.imshow('thresh', thresh)
76     cv2.imshow('frameDiff', frameDiff)
77 
78     # 处理按键效果
79     key = cv2.waitKey(60) & 0xff
80     if key == 27:   # 按下ESC时,退出
81         break
82     elif key == ord(' '):   # 按下空格键时,暂停
83         cv2.waitKey(0)
84 
85 # 释放资源并关闭所有窗口
86 camera.release()
87 cv2.destroyAllWindows()

View Code