Skip to content

Face-similarity CNN using Tensorflow Eager execution.

Notifications You must be signed in to change notification settings

sthalles/face-similarity

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

21 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Face-similarity

Face-similarity CNN using Tensorflow Eager execution.

Medium article: How to train your own FaceID ConvNet using TensorFlow Eager execution

This implementation uses DenseNets with contrastive loss.

Reference Papers:

Dependencies:

  • Python 3.6.x
  • Tensorflow 1.10.1

Inference

Google Colaboratory

  • Just open the inference.ipynb and select the option to open on Colab.

Running locally

1- Download the pre-trained model using the following link.

  • Place the tboard_logs folder in the root folder of the project.

2- Download the following test dataset (TfRecords format).

  • Place the dataset folder in the root folder of the project.

3- Run the jupyter notebook inference.ipynb

  • Run the notebook. Adjust the dataset paths accordingly.

Results

Results