From 3a80490c2efc942c502f7e0f57828652e497e77c Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Wed, 26 Mar 2025 17:35:33 -0600 Subject: [PATCH] Simplify normalization --- frigate/embeddings/onnx/{facenet.py => face_embedding.py} | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) rename frigate/embeddings/onnx/{facenet.py => face_embedding.py} (96%) diff --git a/frigate/embeddings/onnx/facenet.py b/frigate/embeddings/onnx/face_embedding.py similarity index 96% rename from frigate/embeddings/onnx/facenet.py rename to frigate/embeddings/onnx/face_embedding.py index 3439620a0..0b808f716 100644 --- a/frigate/embeddings/onnx/facenet.py +++ b/frigate/embeddings/onnx/face_embedding.py @@ -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)