kuzmoyev / face-recognition-in-movies

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Face recognition in movies

Task

Implement console application for recognition of actors' faces in the movies. Analyse methods of face recognition in face_recognition library. Application will primarily use IMDB actors faces dataset and TMDB API for retrieving actors faces and movie casts.

Output of the application will be:

  • copy of the input video with highlighted and named faces
  • actors' appearances throughout the movie in csv format
  • visualization of actors appearances throughout the movie

Report will also containe a description of the algorithms and methods used in face_recognition library.

Installation

Requires

  • Python3.6+
  • virtualenv

Setup

git clone git@github.com:kuzmoyev/face-recognition-in-movies.git FaceRecognition
cd FaceRecognition
virtualenv -p /usr/bin/python3 venv
source venv/bin/activate
pip install -r requirements.txt
cd src

Ask for help

./movie_face_recognition.py --help

Quick test

Following command will run analysis on Reservoir Dogs trailer. It will download casts photos to imdb/imdb_data/tmdb_images.

./movie_face_recognition.py "test_data/Reservoir Dogs.mp4" -fd

Should run at ~10 fps. It analyses 4 out of 20 frames per second (every 5th frame) hence those jumps.

About


Languages

Language:Jupyter Notebook 55.1%Language:Python 25.6%Language:TeX 18.3%Language:Shell 1.0%