github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-04 02:20:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

bug in how.py sorted

Browse Source
This commit is contained in:
Sefik Ilkin Serengil 2023-01-31 20:58:09 +00:00
parent 6c5c3511f4
commit 9fed010762
1 changed files with 12 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
tests/face-recognition-how.py
View File

@ -1,30 +1,28 @@
import matplotlib.pyplot as plt
import numpy as np
from deepface.basemodels import VGGFace
from deepface import DeepFace
from deepface.commons import functions
# ----------------------------------------------
# build face recognition model
model = VGGFace.loadModel()
model_name = "VGG-Face"
try:
input_shape = model.layers[0].input_shape[1:3]
except: # issue 470
input_shape = model.layers[0].input_shape[0][1:3]
model = DeepFace.build_model(model_name=model_name)
print("model input shape: ", model.layers[0].input_shape[1:])
print("model output shape: ", model.layers[-1].input_shape[-1])
target_size = functions.find_target_size(model_name)
print(f"target_size: {target_size}")
# ----------------------------------------------
# load images and find embeddings
# img1 = functions.detectFace("dataset/img1.jpg", input_shape)
img1 = functions.preprocess_face("dataset/img1.jpg", input_shape)
img1 = DeepFace.extract_faces(img_path="dataset/img1.jpg", target_size=target_size)[0]["face"]
img1 = np.expand_dims(img1, axis=0) # to (1, 224, 224, 3)
img1_representation = model.predict(img1)[0, :]
# img2 = functions.detectFace("dataset/img3.jpg", input_shape)
img2 = functions.preprocess_face("dataset/img3.jpg", input_shape)
img2 = DeepFace.extract_faces(img_path="dataset/img3.jpg", target_size=target_size)[0]["face"]
img2 = np.expand_dims(img2, axis=0)
img2_representation = model.predict(img2)[0, :]
# ----------------------------------------------
@ -58,7 +56,7 @@ distance_graph = np.array(distance_graph)
fig = plt.figure()
ax1 = fig.add_subplot(3, 2, 1)
plt.imshow(img1[0][:, :, ::-1])
plt.imshow(img1[0])
plt.axis("off")
ax2 = fig.add_subplot(3, 2, 2)
@ -66,7 +64,7 @@ im = plt.imshow(img1_graph, interpolation="nearest", cmap=plt.cm.ocean)
plt.colorbar()
ax3 = fig.add_subplot(3, 2, 3)
plt.imshow(img2[0][:, :, ::-1])
plt.imshow(img2[0])
plt.axis("off")
ax4 = fig.add_subplot(3, 2, 4)