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

added 3D batch processing compatibility, patched bugs #196

Closed jklewis99 closed 4 years ago

jklewis99 commented 4 years ago

I was working with @imadtoubal on research and we were able to take advantage of the batch processing with BlazeFace and face_alignment for a speedup in processing. @esterivera noticed our batch processing was incompatible with 3D landmarks. I have now made this available. Additionally, if using the SFD face detector, batch processing would have thrown a RuntimeError: "value cannot be converted to type uint8_t without overflow." I have patched this bug. Additionally, if using setting the flip_input argument to True, we had an undefined variable. I made this correction as well. Hope this helps.

1adrianb commented 4 years ago

Hi @jklewis99 thanks a lot for your contribution! Could you make sure that the linter is happy and address the travis build failure? Thanks again!