用PYTHON写一个通过人脸识别,并根据识别结果将相同人脸照片进行分别保存的软件
时间: 2024-03-05 19:51:23 浏览: 146
好的,这是一个比较复杂的问题,需要用到多个 Python 库。首先,需要安装 OpenCV 库和 face_recognition 库,用于图像处理和人脸识别。然后,我们可以编写一个 Python 脚本,来实现这个功能。
以下是基本代码框架:
```python
import cv2
import face_recognition
import os
# 加载已知人脸的图片
known_face_encodings = []
known_face_names = []
for file_name in os.listdir("known_faces"):
image = face_recognition.load_image_file("known_faces/" + file_name)
encoding = face_recognition.face_encodings(image)[0]
name = file_name.split(".")[0]
known_face_encodings.append(encoding)
known_face_names.append(name)
# 创建一个窗口显示摄像头
cv2.namedWindow("Face Recognition")
# 打开摄像头
video_capture = cv2.VideoCapture(0)
while True:
# 读取摄像头捕捉的画面
ret, frame = video_capture.read()
# 将捕捉到的画面缩小一倍,加快处理速度
small_frame = cv2.resize(frame, (0, 0), fx=0.5, fy=0.5)
# 转换为 RGB 格式
rgb_small_frame = small_frame[:, :, ::-1]
# 在捕捉到的画面中寻找人脸
face_locations = face_recognition.face_locations(rgb_small_frame)
face_encodings = face_recognition.face_encodings(rgb_small_frame, face_locations)
# 遍历捕捉到的所有人脸
for (top, right, bottom, left), face_encoding in zip(face_locations, face_encodings):
# 对比已知人脸
matches = face_recognition.compare_faces(known_face_encodings, face_encoding)
# 找到最相似的人脸
face_distances = face_recognition.face_distance(known_face_encodings, face_encoding)
best_match_index = np.argmin(face_distances)
if matches[best_match_index]:
name = known_face_names[best_match_index]
else:
name = "Unknown"
# 在画面中标注出人脸和姓名
cv2.rectangle(frame, (left*2, top*2), (right*2, bottom*2), (0, 0, 255), 2)
cv2.putText(frame, name, (left*2, bottom*2), cv2.FONT_HERSHEY_DUPLEX, 1.0, (255, 255, 255), 1)
# 根据姓名创建文件夹,并将相同人脸照片保存到对应的文件夹中
if name != "Unknown":
if not os.path.exists(name):
os.makedirs(name)
cv2.imwrite(name + "/" + str(time.time()) + ".jpg", frame)
# 显示处理后的画面
cv2.imshow("Face Recognition", frame)
if cv2.waitKey(1) & 0xFF == ord('q'):
break
# 释放摄像头并关闭窗口
video_capture.release()
cv2.destroyAllWindows()
```
这个脚本会不断读取摄像头捕捉的画面,检测其中的人脸,并进行识别。如果识别到已知人脸,就会在画面中标注出人脸和姓名,并将相同人脸的照片保存到对应的文件夹中。如果识别到未知人脸,则不进行保存。
需要注意的是,这个脚本只是一个基本的框架,可能需要根据具体的需求进行修改和完善。例如,可以增加对多人脸的处理,或者将识别结果输出到数据库中。
阅读全文