/siamese_network_LFW

Primary LanguagePython

Siamese network for LFW

Data generation

  1. Download the data from here and extract it to 'data' folder
  2. Run the following command from the root directory: python3 generate_data.py

A 'images_list.txt' file will be created in 'data' folder.

Model Details

TO DO

Training

  1. Run the following command:
    python3 main.py
  2. The training will begin and model files will be saved in 'snapshot' folder. A best model file will be saved based on the best validation accuracy achieved.

Model

A pretrained model can be downloaded from here.

Inference

  1. For sanity check, run the following command:
    python3 inference.py
  2. You will get an output as follows: alt test

TO DO

Analyse the predicted output