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

dlib true-batched forward

Browse Source
This commit is contained in:
galthran-wq 2025-02-11 17:25:32 +00:00
parent a4a579e5eb
commit 8becc97512
1 changed files with 8 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
deepface/models/facial_recognition/Dlib.py
View File

@ -40,21 +40,17 @@ class DlibClient(FacialRecognition):
if len(img.shape) == 3:
img = np.expand_dims(img, axis=0)
embeddings = []
for single_img in img:
# bgr to rgb
single_img = single_img[:, :, ::-1] # bgr to rgb
# bgr to rgb
img = img[:, :, :, ::-1] # bgr to rgb
# img is in scale of [0, 1] but expected [0, 255]
if single_img.max() <= 1:
single_img = single_img * 255
# img is in scale of [0, 1] but expected [0, 255]
if img.max() <= 1:
img = img * 255
single_img = single_img.astype(np.uint8)
img = img.astype(np.uint8)
img_representation = self.model.model.compute_face_descriptor(single_img)
img_representation = np.array(img_representation)
embeddings.append(img_representation.tolist())
embeddings = self.model.model.compute_face_descriptor(img)
embeddings = [np.array(embedding).tolist() for embedding in embeddings]
if len(embeddings) == 1:
return embeddings[0]
else: