nik/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-08-31 07:27:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Rename to arcface

Browse Source
This commit is contained in:
Nicolas Mowen 2025-03-25 14:40:14 -06:00
parent 3b77e02a40
commit d1767592cb
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
frigate/data_processing/common/face/model.py
View File

@ -8,7 +8,7 @@ from scipy import stats
from frigate.config import FrigateConfig
from frigate.const import MODEL_CACHE_DIR
from frigate.embeddings.onnx.facenet import FaceNetEmbedding
from frigate.embeddings.onnx.facenet import ArcfaceEmbedding
logger = logging.getLogger(__name__)
@ -202,7 +202,7 @@ class ArcFaceRecognizer(FaceRecognizer):
def __init__(self, config: FrigateConfig):
super().__init__(config)
self.mean_embs: dict[int, np.ndarray] = {}
self.face_embedder: FaceNetEmbedding = FaceNetEmbedding()
self.face_embedder: ArcfaceEmbedding = ArcfaceEmbedding()
def clear(self) -> None:
self.mean_embs = {}

4
frigate/embeddings/onnx/facenet.py
View File

@ -16,7 +16,7 @@ logger = logging.getLogger(__name__)
FACE_EMBEDDING_SIZE = 112
class FaceNetEmbedding(BaseEmbedding):
class ArcfaceEmbedding(BaseEmbedding):
def __init__(
self,
device: str = "AUTO",
@ -25,7 +25,7 @@ class FaceNetEmbedding(BaseEmbedding):
model_name="facedet",
model_file="facenet.onnx",
download_urls={
"facenet.onnx": "https://github.com/NickM-27/facenet-onnx/releases/download/v1.0/facenet.onnx",
"arcface.onnx": "https://github.com/NickM-27/facenet-onnx/releases/download/v1.0/arcface.onnx",
},
)
self.device = device