github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-07 12:05:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

use GPU in fastMtCnn

Browse Source
This commit is contained in:
aku-ato 2024-03-26 11:12:54 +01:00 committed by GitHub
parent 329606ffe8
commit 3d8733812b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
deepface/detectors/FastMtCnn.py
View File

@ -68,7 +68,7 @@ class FastMtCnnClient(Detector):
"Please install using 'pip install facenet-pytorch' " "Please install using 'pip install facenet-pytorch' "
) from e ) from e
face_detector = fast_mtcnn(device="cpu") face_detector = fast_mtcnn(device="cuda:0")
return face_detector return face_detector