how to optimize and speed up the alignment code?

[Jar7]

Hello, thank your for your work.
I read the alignment code and found there is a pytorch version alignment function. Can it be used for inference? Do you think it is possible to align faces by batch or any other ways to speed up the alignment process? like run in gpu?
Thank you!

[vitoralbiero]

Hello @Jar7,
Thanks for your interest in our work!

Right now on the run_alignment.py script, inference is running one image at a time, but you could easily modify it to run on a batch of images and then perform alignment on each one.
I haven't measured the time necessary for the alignment part of the script, but I believe it to be negligible compared to the inference time.
If you decide to make improvements, PRs are welcome!

Hope this helps.