首先,我们需要安装Python和OpenCV库。前者可以在官网下载安装程序,后者可以使用pip命令进行安装,在命令行中运行以下命令即可:
pip install opencv-python
接下来的步骤,我们将分为以下三个部分进行:
1. 读取图像并进行人脸检测
我们将使用OpenCV中的CascadeClassifier
类来检测人脸。在此之前,我们需要下载人脸检测器的分类器文件(.xml)并将其放置在项目目录中。
import cv2
# 加载人脸检测器分类器文件
face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
# 读取图像
img = cv2.imread('image.jpg')
# 转换为灰度图像
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
# 进行人脸检测
faces = face_cascade.detectMultiScale(gray, 1.3, 5)
# 在图像上标记出人脸
for (x,y,w,h) in faces:
cv2.rectangle(img,(x,y),(x+w,y+h),(255,0,0),2)
# 显示图像
cv2.imshow('img',img)
cv2.waitKey(0)
cv2.destroyAllWindows()
在上述代码中,我们首先加载了人脸检测器分类器文件,并读取了一张图像。然后,我们将图像转换为灰度图像,这是因为人脸检测器通常需要处理灰度图像来提高检测率。接下来,我们使用CascadeClassifier
类的detectMultiScale()
方法检测人脸,并将检测结果保存在一个元组中。最后,我们在原图像上使用rectangle()
方法标记出检测到的人脸,并将其显示出来。
2. 成功检测人脸后对图像进行操作
在检测到人脸后,我们可以根据具体需求对图像进行操作。例如,我们可以在人脸周围添加一个矩形框,并将处理后的图像保存到本地。
import cv2
# 加载人脸检测器分类器文件
face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
# 读取图像
img = cv2.imread('image.jpg')
# 转换为灰度图像
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
# 进行人脸检测
faces = face_cascade.detectMultiScale(gray, 1.3, 5)
# 在图像上标记出人脸
for (x,y,w,h) in faces:
cv2.rectangle(img,(x,y),(x+w,y+h),(255,0,0),2)
# 保存处理后的图像
cv2.imwrite('result.jpg', img)
在以上代码中,我们将原图像中检测到的人脸周围添加了一个蓝色矩形框,并将处理后的图像保存到了本地。
3. 检测并保存视频中的人脸
在检测和处理图像的基础上,我们还可以将其应用于视频中。下面的代码可以读取视频文件,检测并保存其中的人脸。
import cv2
# 加载人脸检测器分类器文件
face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
# 打开摄像头或者读取视频文件
cap = cv2.VideoCapture(0)
# 定义视频编码器及输出文件名称
fourcc = cv2.VideoWriter_fourcc('M', 'J', 'P', 'G')
out = cv2.VideoWriter('output.avi', fourcc, 10.0, (640,480))
while True:
# 读取一帧图像
ret, frame = cap.read()
# 转换为灰度图像
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
# 进行人脸检测
faces = face_cascade.detectMultiScale(gray, 1.3, 5)
# 在图像上标记出人脸
for (x,y,w,h) in faces:
cv2.rectangle(frame,(x,y),(x+w,y+h),(255,0,0),2)
# 显示图像
cv2.imshow('frame',frame)
# 将处理后的图像保存到视频文件中
out.write(frame)
# 按下q键退出循环
if cv2.waitKey(1) & 0xFF == ord('q'):
break
# 释放资源
cap.release()
out.release()
cv2.destroyAllWindows()
注意,在以上代码中,我们使用了VideoWriter
类来将处理后的图像保存为视频文件。运行程序时,按下q
键可以退出循环。
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:Python基于OpenCV实现人脸检测并保存 - Python技术站