From 3d8733812b7b5db1fcfd38fec95b739d9d6e11c2 Mon Sep 17 00:00:00 2001 From: aku-ato <111570992+aku-ato@users.noreply.github.com> Date: Tue, 26 Mar 2024 11:12:54 +0100 Subject: [PATCH] use GPU in fastMtCnn --- deepface/detectors/FastMtCnn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepface/detectors/FastMtCnn.py b/deepface/detectors/FastMtCnn.py index acc71f1..2df5c4d 100644 --- a/deepface/detectors/FastMtCnn.py +++ b/deepface/detectors/FastMtCnn.py @@ -68,7 +68,7 @@ class FastMtCnnClient(Detector): "Please install using 'pip install facenet-pytorch' " ) from e - face_detector = fast_mtcnn(device="cpu") + face_detector = fast_mtcnn(device="cuda:0") return face_detector