diff --git a/frigate/data_processing/common/face/model.py b/frigate/data_processing/common/face/model.py index 56eda44919..a006734cea 100644 --- a/frigate/data_processing/common/face/model.py +++ b/frigate/data_processing/common/face/model.py @@ -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 = {} diff --git a/frigate/embeddings/onnx/facenet.py b/frigate/embeddings/onnx/facenet.py index 6c44b8f590..0d6155165d 100644 --- a/frigate/embeddings/onnx/facenet.py +++ b/frigate/embeddings/onnx/facenet.py @@ -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