Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-08-18 13:47:20 +02:00
Code Issues 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 6ac2f7f91e
commit 81525e984e
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

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