search

1adrianb / face-alignment

:fire: 2D and 3D Face alignment library build using pytorch
https://www.adrianbulat.com
BSD 3-Clause "New" or "Revised" License
6.88k stars 1.33k forks source link

Add Apple M1 GPU Option to Class 'DlibDetector' #352

Open KennardWang opened 8 months ago

KennardWang commented 8 months ago

In face_alignment/detection/dlib/dlib_detector.py, Apple M1 GPU (device = 'mps') seems to be ignored in the previous version.