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

Conditional use of GPU on FastMtCnn

Browse Source
This commit is contained in:
aku-ato 2024-03-26 18:06:53 +01:00 committed by GitHub
parent 2b7c0a708e
commit 345b6717e2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
deepface/detectors/FastMtCnn.py
View File

@ -2,6 +2,7 @@ from typing import Any, Union, List
import cv2 import cv2
import numpy as np import numpy as np
from deepface.models.Detector import Detector, FacialAreaRegion from deepface.models.Detector import Detector, FacialAreaRegion
import torch
# Link -> https://github.com/timesler/facenet-pytorch # Link -> https://github.com/timesler/facenet-pytorch
# Examples https://www.kaggle.com/timesler/guide-to-mtcnn-in-facenet-pytorch # Examples https://www.kaggle.com/timesler/guide-to-mtcnn-in-facenet-pytorch