mirror of
https://github.com/blakeblackshear/frigate.git
synced 2025-08-13 13:47:36 +02:00
ArcFace uses GPU by default, no need for 'CPU' value.
This commit is contained in:
parent
6ac2f7f91e
commit
81525e984e
@ -149,11 +149,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):
|
||||
|
Loading…
Reference in New Issue
Block a user