Kpop-Idol-Face-Classifier
Kpop-Idol-Face-Classifier copied to clipboard
A face recognition model to recognize your favorite kpop artists.
Kpop Idol Face Classifier
This is a face recognition model aimed at classifying your favourite Kpop singers.
Uses a FaceNet + SVM model to train and classify faces.
The model currently supports recognizing Arin, Chaeyong, Chorong, Dahyun, Eunha, Irene, Jihyo, Lisa, Naeun, Nayeon, Sana, Seulgi, Umji, Yeji and Yuju.
,
Requirements
- Keras 2.3.1 ( No need to have a GPU )
- dlib
- sklearn
- numpy
- OpenCV
Install the above (preferably in a conda environment) and download the data.
Download the FaceNet model from here, and place it inside the model folder.
Data
You can use your own dataset, where each person should have at least 10-15 images.
The directory structure should look like this
- data
- raw_data
- Person_1
- pic1.jpg
- pic2.jpg
- Person 2
- pic1.jpg
- pic2.jpg
- Person_1
- raw_data
Now run process.py, which will crop the faces and put them in the 'data/processed_data/train' folder.
Usage
To use the model as it is, put your images in the 'play' folder and run 'python main.py'.
To add support for more artists, Download the photos and put them under the raw_data folder. Run 'python process.py' and then 'python train.py'.
Cheers!
Samples