学习来自《Learning OpenCV 3 Computer Vision with Python》Second Edition by Joe Minichino and Joseph Howse
文章目录
- 检测移动的目标
- 涉及到的 opencv 库
- cv2.GaussianBlur
- cv2.absdiff
- cv2.threshold
- cv2.dilate
- cv2.getStructuringElement
- cv2.findContours
- cv2.contourArea
- cv2.boundingRect
检测移动的目标
目标跟踪:基本的运动检测
一种最直观的方法就是计算帧之间的差异,或者考虑背景帧与其他帧之间的差异
import cv2
import numpy as npes = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (9, 4)) # 我跑的时候用的 (9,9) 圆
background = None
index = 0cap = cv2.VideoCapture("2.mkv")if cap.isOpened():success = True
else:success = Falseprint("fail to open")while(success):success, frame = cap.read()index += 1h, w, c = frame.shape# 第一帧作为背景if background is None:background = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY) background = cv2.GaussianBlur(background, (21, 21), 0)continue# 对噪声进行平滑是为了避免在运动和跟踪时将其检测出来gray_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)gray_frame = cv2.GaussianBlur(gray_frame, (21, 21), 0)diff = cv2.absdiff(background, gray_frame)diff = cv2.threshold(diff, 127, 255, cv2.THRESH_BINARY)[1] # 大于 127 就置为 255# 腐蚀和膨胀也可以用作噪声滤波器diff = cv2.dilate(diff, es, iterations=2)# image, cnts, hierarchy = cv2.findContours(diff.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)cnts, hierarchy = cv2.findContours(diff.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)for c in cnts:if cv2.contourArea(c) < 0.25*h*0.25*w:# if cv2.contourArea(c) < 2500:continue(x, y, w, h) = cv2.boundingRect(c) # 计算矩形的边界框 cv2.rectangle(frame, (x, y), (x+w, y+h), (0, 0, 255), 10)cv2.imshow("contours", frame)cv2.imshow("diff", diff)if cv2.waitKey(1000 // 12) & 0xff == ord("q"):break
cv2.destroyAllWindows()
cap.release()
先看下效果
原视频,720p
cv2.dilate(diff, es, iterations=2)
时 diff 的效果,也即高斯模糊后的当前帧和背景帧差的绝对值膨胀两次后的效果
过滤掉小于 2500 的轮廓时的效果,并以矩形框的形式可视化出来
看起来太敏感了,我们来个粗犷一些的
膨胀 30 次,cv2.dilate(diff, es, iterations=30)
,diff 的效果如下
过滤掉 if cv2.contourArea(c) < 0.25*h*0.25*w:
面积小于 6.25% 的移动区域,轮廓可视化成矩形框如下
还行
技术缺点
- 需要通过提前设置“默认”帧作为背景,在光照变化频繁时就显得不够灵活
涉及到的 opencv 库
cv2.GaussianBlur
高斯模糊
cv2.absdiff
计算两个数组之间或数组与标量之间每个元素的绝对差
cv2.threshold
二值化函数
cv2.dilate
形态学膨胀
cv2.getStructuringElement
得到一个结构元素(卷积核),主要用于后续的腐蚀、膨胀、开、闭等运算
- MORPH_RECT(函数返回矩形卷积核)
- MORPH_CROSS(函数返回十字形卷积核)
- MORPH_ELLIPSE(函数返回椭圆形卷积核)
anchor 表示描点的位置
cv2.findContours
找轮廓
mode:轮廓的模式。
- cv2.RETR_EXTERNAL 只检测外轮廓;
- cv2.RETR_LIST 检测的轮廓不建立等级关系;
- cv2.RETR_CCOMP 建立两个等级的轮廓,上一层为外边界,内层为内孔的边界。如果内孔内还有连通物体,则这个物体的边界也在顶层;
- cv2.RETR_TREE 建立一个等级树结构的轮廓。
method:轮廓的近似方法
- cv2.CHAIN_APPROX_NOME存储所有的轮廓点,相邻的两个点的像素位置差不超过1;
- cv2.CHAIN_APPROX_SIMPLE 压缩水平方向、垂直方向、对角线方向的元素,只保留该方向的终点坐标,例如一个矩形轮廓只需要4个点来保存轮廓信息;
- cv2.CHAIN_APPROX_TC89_L1,cv2.CV_CHAIN_APPROX_TC89_KCOS
contours:返回的轮廓
hierarchy:每条轮廓对应的属性
cv2.contourArea
轮廓面积
cv2.boundingRect
轮廓拟合函数