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

ArcFace uses GPU by default, no need for 'CPU' value.

Browse Source
This commit is contained in:
baudneo 2025-08-01 17:04:29 -06:00
parent f5d4a4bb6d
commit b103ce7e2e
No known key found for this signature in database
GPG Key ID: 51445F2ED08EBC7F
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
frigate/embeddings/onnx/face_embedding.py
View File

@ -151,11 +151,7 @@ class ArcfaceEmbedding(BaseEmbedding):
self.runner = ONNXModelRunner(
os.path.join(self.download_path, self.model_file),
device=self.config.face_recognition.device
or (
"GPU"
if self.config.face_recognition.model_size == "large"
else "CPU"
),
or "GPU",
)
def _preprocess_inputs(self, raw_inputs):