administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-05-05 13:07:44 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Simplify normalization

Browse Source
This commit is contained in:
Nicolas Mowen 2025-03-26 17:35:33 -06:00
parent 53c8aa25cb
commit 3a80490c2e
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
frigate/embeddings/onnx/facenet.py → frigate/embeddings/onnx/face_embedding.py
View File

@ -90,8 +90,7 @@ class ArcfaceEmbedding(BaseEmbedding):
frame[y_center : y_center + og_h, x_center : x_center + og_w] = og frame[y_center : y_center + og_h, x_center : x_center + og_w] = og
# run arcface normalization # run arcface normalization
normalized_image = frame.astype(np.float32) / 255.0 frame = (frame / 127.5) - 1.0
frame = (normalized_image - 0.5) / 0.5
frame = np.transpose(frame, (2, 0, 1)) frame = np.transpose(frame, (2, 0, 1))
frame = np.expand_dims(frame, axis=0) frame = np.expand_dims(frame, axis=0)