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

[Multiple landmarks for image with 2-3 faces] #156

Closed vinayak618 closed 3 years ago

vinayak618 commented 4 years ago

Hi @1adrianb . How to get the multiple landmark predictions for the image with 2-3 faces in it, and how it chooses just single face from all detected faces after 'sfd' to get the landmarks? Like the below output: test

1adrianb commented 4 years ago

It should already work for multiple faces. Can you please check how many faces did sfd detect? (https://github.com/1adrianb/face-alignment/blob/6a7731168dbb1a15f9ecd5fe4c79c992f179a622/face_alignment/api.py#L140)

vinayak618 commented 4 years ago

It's detecting all the faces in the image, but it will end putting the landmarks for last face maybe it's just the landmarks is not getting appended for all the face predictions.

1adrianb commented 4 years ago

thanks, I will have a look towards the end of this week, sorry for the delay