python通过人脸照片查找相同人脸的其他图片

人脸识别

# 概述
# 在Python中，可以通过人脸识别技术来查找相同人脸的图片。这通常涉及到以下几个步骤：人脸检测、特征提取、特征匹配和结果输出。
#
# 人脸检测
# 首先，需要检测图片中的人脸。可以使用OpenCV或dlib等库来实现人脸检测。
#
# Python
# 复制
import cv2

# 加载人脸检测器
face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')

# 读取图片
img = cv2.imread('path_to_image.jpg')
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), (255, 0, 0), 2)

cv2.imshow('Detected  Faces', img)
cv2.waitKey(0)
cv2.destroyAllWindows()
# 特征提取
# 接下来，需要从检测到的人脸中提取特征。可以使用dlib库中的预训练模型来提取人脸特征。
#
# Python
# 复制
import dlib

# 加载人脸检测器和特征提取器
detector = dlib.get_frontal_face_detector()
sp = dlib.shape_predictor('shape_predictor_68_face_landmarks.dat')
facerec = dlib.face_recognition_model_v1('dlib_face_recognition_resnet_model_v1.dat')

# 读取图片
img = dlib.load_rgb_image('path_to_image.jpg')

# 检测人脸
dets = detector(img, 1)

for k, d in enumerate(dets):
    shape = sp(img, d)
    face_descriptor = facerec.compute_face_descriptor(img, shape)
    print(face_descriptor)
# 特征匹配
# 提取特征后，可以使用某种距离度量（如欧氏距离）来比较不同图片中的人脸特征，从而判断是否为同一人。
#
# Python
# 复制
import numpy as np


def compare_faces(face_descriptor1, face_descriptor2):
    return np.linalg.norm(np.array(face_descriptor1) - np.array(face_descriptor2))


# 假设我们已经提取了两张图片的人脸特征
face_descriptor1 = [...]  # 第一张图片的人脸特征
face_descriptor2 = [...]  # 第二张图片的人脸特征

distance = compare_faces(face_descriptor1, face_descriptor2)
if distance < 0.6:  # 阈值可以根据实际情况调整
    print("两张图片中的人脸是同一个人")
else:
    print("两张图片中的人脸不是同一个人")
# 结果输出
# 最后，可以根据特征匹配的结果输出相同人脸的图片。可以将匹配结果存储在一个列表中，并输出这些图片的路径。
#
# Python
# 复制
matched_images = []

# 假设我们有一组图片和它们的人脸特征
image_paths = ['path_to_image1.jpg', 'path_to_image2.jpg', ...]
face_descriptors = [face_descriptor1, face_descriptor2, ...]

for i in range(len(image_paths)):
    for j in range(i + 1, len(image_paths)):
        distance = compare_faces(face_descriptors[i], face_descriptors[j])
        if distance < 0.6:
            matched_images.append((image_paths[i], image_paths[j]))

for pair in matched_images:
    print(f"相同人脸的图片对: {pair[0]} 和 {pair[1]}")
# 通过以上步骤，可以在Python中实现通过人脸查找相同人脸的图片。