Skip to content

pwliuab/MachineLearning-face-similarity-detection

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 

Repository files navigation

MachineLearning-face-similarity-detection

Introduction

- 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 :)

📃 Installation

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

Dataset dowload

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

Data Preprocessing

Deal with the data imbalance

Model Structure

Using CNN

Test with different layers of CNN

Model Training

15 epochs

Customize the training function and load function.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published