用opencv对人脸数据集提取人脸并保存代码
时间: 2023-05-30 18:02:59 浏览: 58
以下是使用OpenCV提取人脸并保存的Python代码:
```python
import cv2
import os
# 加载人脸识别分类器
face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
# 设置图片目录和输出目录
img_dir = 'path/to/image/directory/'
out_dir = 'path/to/output/directory/'
# 遍历图片目录
for filename in os.listdir(img_dir):
if filename.endswith('.jpg') or filename.endswith('.png'):
# 读取图片
img = cv2.imread(os.path.join(img_dir, filename))
# 转换为灰度图像
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
# 检测人脸
faces = face_cascade.detectMultiScale(gray, scaleFactor=1.1, minNeighbors=5, minSize=(30, 30))
# 遍历每个人脸
for (x, y, w, h) in faces:
# 绘制人脸矩形框
cv2.rectangle(img, (x, y), (x+w, y+h), (0, 255, 0), 2)
# 提取人脸并保存
out_filename = os.path.splitext(filename)[0] + '_face.jpg'
out_path = os.path.join(out_dir, out_filename)
face_img = img[y:y+h, x:x+w]
cv2.imwrite(out_path, face_img)
```
在这个代码中,我们首先加载了人脸识别分类器,然后设置了图片目录和输出目录。接着,我们遍历图片目录中的每个图片,读取并转换为灰度图像,然后使用人脸识别分类器检测人脸。对于每个检测到的人脸,我们绘制一个矩形框,并使用矩形框提取人脸图像,最后保存到输出目录中。