问 Pytorch Facenet MTCNN图像输出

EN

我正在用两种方法在python中使用facenet py手电筒( https://github.com/timesler/facenet-pytorch )做人脸识别应用程序。

第一方法 代码-

resnet = InceptionResnetV1(pretrained='vggface2').eval()
mtcnn = MTCNN(image_size=96)
img = Image.open(image_path)
image_prep = mtcnn(img)
plt.figure()
plt.axis('off')
plt.imshow(image_prep.permute(1, 2, 0))
if image_prep is not None:
  image_embedding = resnet(image_prep.unsqueeze(0))

在这段代码中，我从给定的图像中提取人脸，并获得用于识别人脸的512编码。

在这个例子中，我使用了两个不同的面孔，并绘制了两张脸之间的距离。

        a.jpg       b.jpg
a.jpg   0.000000    1.142466
b.jpg   1.142466    0.000000

效果很好..。

第二方法 代码-

 img = Image.open(image)
 boxes, probs = mtcnn.detect(img) # Gives the coordinates of the face in the given image
 face = img.crop((boxes[0][0], boxes[0][1], boxes[0][2], boxes[0][3])) # Cropping the face