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.94k stars 1.33k forks source link

Can retinaface be used instead of sfd? #164

Open Coderx7 opened 4 years ago

Coderx7 commented 4 years ago

Is it possible to use RetinaFace instead of SFD ?

miracleyoo commented 4 years ago

Yes, I succeed in adding RetinaFace into the detector folder and use it instead of SFD. You need to read how the SFD part is written, copy the SFD folder, rename it to retina, and change the init.py, do your replacement in the retina_detector.py file... and so on. Just follow how the SFD is implemented and imitate it.

1adrianb commented 4 years ago

@miracleyoo if you have time, feel free to make a pull request :)

miracleyoo commented 4 years ago

Sure :) Then I need to clean up the code and figure out how to use torch utils load_url since I put the model in my local space directly now lol.

ibadamimarz commented 3 years ago

@miracleyoo Did you get a chance to make a PR for retina ? :) Thank you!