How can I change the code to recognize two person?

[Jin-Yerom]

I am really appreciate your code, and I am interested in this code. I am studying EleGANt and want to give some changes. I found that this EleGANt model cannot recognize 2 person. So I want to get your advice. How can I change the code to recognize two person? WHICH CODE I NEED TO STUDY AND REFORM? THANK YOU.

[Chenyu-Yang-2000]

Thanks for your interest, @Jin-Yerom. We use the dlib library to detect and crop the face. Since all data samples now only contain one person, we simply take the first face recognized by dlib. (Refer to training/process.py). You can check whether the dlib library supports detecting multiple faces and modify the code. i.e., take several items from the list of faces.