/FaceDetection

Viola-Jones on GPU

Primary LanguageJupyter Notebook

Viola-Jones algorithm on GPU

  • This is an implementation of the Viola-Jones algorithm for face detection using PyToch.
  • It uses PyTorch&CUDA to perform matrix operations.
  • Its training time is 8.5 times faster than sklearn.
  • Feature extraction, training, and inference all were implemented using torch to utilize GPU.
  • It was trained on the FDDB dataset.
  • Cropped faces, non-faces can be found here

Alt text

Alt text

Alt text

Alt text