用python编写一个人脸识别的脚本
时间: 2023-09-17 07:13:07 浏览: 79
好的,我可以为您提供一个使用Python编写的简单人脸识别脚本,但需要注意的是,这仅仅是一个示例代码,仅用于学习和参考,如果需要用于实际应用中,需要根据具体情况进行改进和优化。
首先,需要安装一些必要的库,包括OpenCV和face_recognition。您可以使用pip命令来安装它们:
```
pip install opencv-python
pip install face_recognition
```
然后,您需要准备一些人脸图像作为训练集和测试集。将它们放在两个不同的文件夹中,例如“train”和“test”。
接下来,我们可以编写代码了。以下是一个简单的示例:
```python
import cv2
import face_recognition
import os
# 加载训练集中的人脸图像
train_dir = "train"
train_images = []
train_labels = []
for name in os.listdir(train_dir):
img_path = os.path.join(train_dir, name)
img = face_recognition.load_image_file(img_path)
encoding = face_recognition.face_encodings(img)[0]
train_images.append(encoding)
train_labels.append(name.split(".")[0])
# 加载测试集中的人脸图像
test_dir = "test"
for name in os.listdir(test_dir):
img_path = os.path.join(test_dir, name)
img = cv2.imread(img_path)
img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
# 在测试图像中查找人脸
face_locations = face_recognition.face_locations(img_rgb)
face_encodings = face_recognition.face_encodings(img_rgb, face_locations)
# 对于每个人脸,进行识别并在图像中标出
for face_encoding, face_location in zip(face_encodings, face_locations):
matches = face_recognition.compare_faces(train_images, face_encoding)
name = "Unknown"
# 找到最相似的人脸
face_distances = face_recognition.face_distance(train_images, face_encoding)
best_match_index = np.argmin(face_distances)
if matches[best_match_index]:
name = train_labels[best_match_index]
# 在人脸周围画一个框,并显示姓名
top, right, bottom, left = face_location
cv2.rectangle(img, (left, top), (right, bottom), (0, 0, 255), 2)
cv2.putText(img, name, (left + 6, bottom - 6), cv2.FONT_HERSHEY_DUPLEX, 1.0, (255, 255, 255), 1)
# 显示识别结果
cv2.imshow(name, img)
cv2.waitKey(0)
```
在这个示例中,我们首先加载训练集中的人脸图像,并使用face_recognition库将它们转换为特征向量。然后,我们加载测试集中的人脸图像,并使用OpenCV库在图像中查找人脸。对于每个人脸,我们使用face_recognition库将其转换为特征向量,并与训练集中的人脸进行比较,找到最相似的人脸。最后,我们在图像中标出人脸,并显示其姓名。
请注意,这只是一个简单的示例代码,您可以根据实际情况进行改进和优化,例如使用深度学习模型来提高识别准确率。同时,这个脚本还存在一些问题,例如当测试集中出现多个人脸时,可能会出现混淆的情况。因此,在实际应用中,需要根据具体情况进行改进和优化。
阅读全文