github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-07 12:05:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

VGGFace batched inference

Browse Source
This commit is contained in:
galthran-wq 2025-02-11 17:00:59 +00:00
parent c60152e9a5
commit 8fb70eb43f
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
deepface/models/facial_recognition/VGGFace.py
View File

@ -57,8 +57,7 @@ class VggFaceClient(FacialRecognition):
def forward(self, img: np.ndarray) -> List[float]: def forward(self, img: np.ndarray) -> List[float]:
""" """
Generates embeddings using the VGG-Face model. Generates embeddings using the VGG-Face model.
This method incorporates an additional normalization layer, This method incorporates an additional normalization layer.
necessitating the override of the forward method.
Args: Args:
img (np.ndarray): pre-loaded image in BGR img (np.ndarray): pre-loaded image in BGR
@ -70,8 +69,14 @@ class VggFaceClient(FacialRecognition):
# having normalization layer in descriptor troubles for some gpu users (e.g. issue 957, 966) # having normalization layer in descriptor troubles for some gpu users (e.g. issue 957, 966)
# instead we are now calculating it with traditional way not with keras backend # instead we are now calculating it with traditional way not with keras backend
embedding = self.model(img, training=False).numpy()[0].tolist() embedding = super().forward(img)
embedding = verification.l2_normalize(embedding) if (
isinstance(embedding, list) and
isinstance(embedding[0], list)
):
embedding = verification.l2_normalize(embedding, axis=1)
else:
embedding = verification.l2_normalize(embedding)
return embedding.tolist() return embedding.tolist()