Skip to content
This repository has been archived by the owner on Jul 17, 2019. It is now read-only.

Siamese Network and Triplet Loss for face recognition in real time

Notifications You must be signed in to change notification settings

pwz266266/SiameseNetwork-pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 

Repository files navigation

This repo is archived due to open source version of the original project will be uploaded and maintained.

This file is used to track progress of face recognition part.

  • The prototype is still buggy and need to be fixed.
  • use command python main.py to run face recognition with siamese network prototype.
  • Dataset available from https://github.com/StephenMilborrow/muct.git
  • New dataset available from http://mmlab.ie.cuhk.edu.hk/projects/CelebA.html
  • Need to create a trainset/testset/validset csv/txt file before training which include names of all images.
  • File format as IMAGE_NAME IMAGE_LABEL in each line
  • Network contains 11 conv + 4 pooling + 2 maxout + 2 fc layers.

If program crashes, try to reduce BATCH_SIZE, require roughly 11G memory for BATCH_SIZE=130.

TODO

  • Add validation step
  • Improving algorithm efficiency
  • Maximize GPU usage
  • Add comment
  • Add functionality to save and load trained model
  • Rewrite part of code for easy customizing
  • Rewrite according to given API

Dependency

  • python 3.*
  • pytorch 0.4.1
  • torchvision
  • matplotlib
  • skimage
  • numpy
  • pandas

Statement

  • The origin of this project is from one of my second year module in University of Nottingham.
  • That project focus on build a real time face recognition application for low quality video/CCTV.
  • This is face recognition part of that origin project, all writen by author.

About

Siamese Network and Triplet Loss for face recognition in real time

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages