项目作者: luickk

项目描述 :
Face identification based on CNN and OpenCv face detection
高级语言: Python
项目地址: git://github.com/luickk/cnn-identity-classification.git
创建时间: 2017-10-19T09:57:39Z
项目社区:https://github.com/luickk/cnn-identity-classification

开源协议:

下载


Face Identity Classification

The project aims to recognizes and identify faces in pictures by using opencv to detect faces and a keras CNN to identify them.


Installation

  1. git clone https://github.com/luickk/cnn-face-recognition

Training

Install listed dependiencies

  • numpy
  • Keras
  • cv2
  • Pillow

Then create the data dir

  1. mkdir data

Add your data to data/data/class/imgFiles

Used dataset: https://www.kaggle.com/dansbecker/5-celebrity-faces-dataset

  1. python3 train.py

Testing

Fill in trained model uid and test-img path beforehand

  1. python predict.py

Results

img_classified

Original:

img_org