Face recognition bug fixes (#17401)

* Simplify normalization

* Fix confidence check
This commit is contained in:
Nicolas Mowen
2025-03-26 19:43:10 -05:00
committed by GitHub
parent a37f804469
commit ff34739f40
2 changed files with 3 additions and 4 deletions
@@ -90,8 +90,7 @@ class ArcfaceEmbedding(BaseEmbedding):
frame[y_center : y_center + og_h, x_center : x_center + og_w] = og
# run arcface normalization
normalized_image = frame.astype(np.float32) / 255.0
frame = (normalized_image - 0.5) / 0.5
frame = (frame / 127.5) - 1.0
frame = np.transpose(frame, (2, 0, 1))
frame = np.expand_dims(frame, axis=0)