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
7.11k stars 1.35k forks source link

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

Open KennardWang opened 1 year ago

KennardWang commented 1 year ago

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