最近在做个机器人比赛,其中一项要求是让机器人实现对是否有人员活动的检测,所以就先拿PC端写一下,准备移植到机器人的树莓派。
(1).我们开启摄像头模块还是利用openCV,如下代码即可。我们使用的是笔记本电脑自带的摄像头(树莓派想直接用一个USB摄像头插上而不是用CSI摄像头,这样直接移植代码比较方便)
camera = cv2.VideoCapture(0)
(2).查看一下camera的尺寸,用以下两个函数
width = int(camera.get(cv2.CAP_PROP_FRAME_WIDTH))
height = int(camera.get(cv2.CAP_PROP_FRAME_HEIGHT))
(3).检测轮廓还是需要用灰度图像,不明白的可以看看我以前的文章
gray_pic = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
上一步骤即把图像灰度化
(4).重点说一下这个FPS的设置,我用以下写法,也是参考了一下别人的代码哈=、=
start = time.time()
"""
xxxxx中间一部分代码省略
"""
seconds = end - start
if seconds < 1.0 / fps:
time.sleep(1.0 / fps - seconds)
(5).进行一下高斯滤波模糊处理
这里说一下高斯滤波是一种线性平滑滤波,通俗的讲,高斯滤波就是对整幅图像进行加权平均的过程,每一个像素点的值,都由其本身和邻域内的其他像素值经过加权平均后得到。我理解的就是一个卷积的过程。
听着很麻烦但实现很简单
gray_pic = cv2.GaussianBlur(gray_pic, (21, 21), 0)
注意(21,21)中必须是奇数,否则会出错,想具体了解的话有时间单独写一篇关于高斯的文章
(6).计算两图片的差
img_delta = cv2.absdiff(pre_frame, gray_pic)
(7).滤波,这个代码的意思是把灰度图中33以下的归为0,33以上的点归为255
thresh = cv2.threshold(img_delta, 30, 255, cv2.THRESH_BINARY)[1]
(8).利用形态学的腐蚀与膨胀(上篇文章讲过了,有时间详细说一下,网上资料也很多)
thresh = cv2.dilate(thresh, None, iterations=2)
(9).findContours检测物体的轮廓
image, contours, hierarchy = cv2.findContours(thresh.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
(10).在while(1)循环中结束摄像头的调用(按Q结束)
大家都在用这种方法,直接Copy就好啦
if cv2.waitKey(1) & 0xFF == ord('q'):
break
import cv2
import time
# 定义摄像头对象,其参数0表示第一个摄像头
camera = cv2.VideoCapture(0)
# 测试用,查看视频size
width = int(camera.get(cv2.CAP_PROP_FRAME_WIDTH))
height = int(camera.get(cv2.CAP_PROP_FRAME_HEIGHT))
size = width,height
#打印一下分辨率
print(repr(size))
#设置一下帧数和前背景
fps = 5
pre_frame = None
while (1):
start = time.time()
# 读取视频流
ret, frame = camera.read()
# 转灰度图
gray_pic = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
if not ret:
print("打开摄像头失败")
break
end = time.time()
cv2.imshow("capture", frame)
# 运动检测部分,看看是不是5FPS
seconds = end - start
if seconds < 1.0 / fps:
time.sleep(1.0 / fps - seconds)
gray_pic = cv2.resize(gray_pic, (480, 480))
# 用高斯滤波进行模糊处理
gray_pic = cv2.GaussianBlur(gray_pic, (21, 21), 0)
# 如果没有背景图像就将当前帧当作背景图片
if pre_frame is None:
pre_frame = gray_pic
else:
# absdiff把两幅图的差的绝对值输出到另一幅图上面来
img_delta = cv2.absdiff(pre_frame, gray_pic)
# threshold阈值函数(原图像应该是灰度图,对像素值进行分类的阈值,当像素值高于(有时是小于)阈值时应该被赋予的新的像素值,阈值方法)
thresh = cv2.threshold(img_delta, 30, 255, cv2.THRESH_BINARY)[1]
# 用一下腐蚀与膨胀
thresh = cv2.dilate(thresh, None, iterations=2)
# findContours检测物体轮廓(寻找轮廓的图像,轮廓的检索模式,轮廓的近似办法)
image, contours, hierarchy = cv2.findContours(thresh.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
for c in contours:
# 设置敏感度
# contourArea计算轮廓面积
if cv2.contourArea(c) < 1000:
continue
else:
print("有人员活动!!!")
# 保存图像
TI = time.strftime('%Y-%m-%d', time.localtime(time.time()))
cv2.imwrite("D:\\PYthon\\first_j\\" + "JC"+TI+ '.jpg', frame)
break
pre_frame = gray_pic
if cv2.waitKey(1) & 0xFF == ord('q'):
break
# release()释放摄像头
camera.release()
# destroyAllWindows()关闭所有图像窗口
cv2.destroyAllWindows()
开启后动了一下脸,就可以看到输出啦,并且把照片已经保存了下来
效果图在下面,就不暴露帅气的外表了
效果还可以,不动的话不会拍照,如果想让他愚钝点可以调节灵敏度
新手一个,如果有错误欢迎指正交流讨论,转载请注明出处。
QQ:78034835