- Detect whether two faces are similar
- You may input your family photos, make comparsion! See whether you two are alike !
- You may even Customize your own model, improve your model based on the current coding :)
Open it(pa2) using google colab, run it using GPU resources or you may use jupyter note book to open.
Please follow the comment inside the notebook file
Train.csv - validate.csv storing all the information of image directory.
Click the google drive to download: https://drive.google.com/file/d/1H4WEqCbCutkTXfOOHd66MaANw9bs-Onu/view?usp=sharing
Deal with the data imbalance
Using CNN
Test with different layers of CNN
15 epochs
Customize the training function and load function.