The repo has 3 folders -
- FaceNet - Contains Python Notebooks for Generating the cropped images, actual training, creating the embeddings and generating the predictions. Also contains Python Notebook for Evaluating the predictions and generating the classification report and the confusion matrix.
- MobileNetV3 - Contains Python Notebooks for pre-processing the data, augmenting it, and for actually training the MobileNetV3 (from scratch and transfer learning + finetuning)
- Webpage - https://github.com/AmoghDabholkar/CS6220_Project/tree/main/Webpage#website-for-cs6220-group-project