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)