Skip to content

Latest commit

 

History

History
177 lines (126 loc) · 6.51 KB

README.md

File metadata and controls

177 lines (126 loc) · 6.51 KB
Raw

Face Recognition Using Cascade Detectors & Facial Expression Classification Using CNN

Notes

  • The datasets are not pushed to the repo!
  • The result data (cascade xml file and cnn models) are not pushed to the repo!

Stages

  1. Gathering Dataset
  2. Train Cascade Classifier
  3. Train CNN
  4. Merge Face Detection and Facial Expression Recognition

Gathering Dataset

We need 2 datasets for our project:

  1. Dataset for Cascade Classifier
  2. Dateset for CNN

The first dataset is used to detect face among other objects.
So the faces are the positive data and other objects could be the negative.

The second dataset is used for CNN feed. After the CNN training, the trained model can recognize five different emotions. Therefore, we seperate the second dataset to the following expressions. \

  • neutral
  • happy
  • sad
  • surprised
  • angry

You can train the cascade classifier and the CNN with any type of data!

Train Cascade Classifier

You can find relevant code in cascade folder of the repository.

To be honest, this part of job was a little tricky and boring.
To use the haar or lbp opencv_traincascade, first you should:

  1. Get opencv and opencv_contrib command collections
  2. Run the command via cli rather than python program

So we had decided to write some bash scripts to make the process easier.
This is what I'd wrote as an Instruction:

  1. fill out the folders: "neg_orig" and "pos_orig" with images
  2. run to_gray.python program. It resizes and converts images to grayscale and puts them to "pos" and "neg" folders
  3. run create_bg.sh and create_info.sh to create information files: "bg.txt" for negatives and "info.dat" for positives (information files init)
  4. run create_neg.sh and create_pos.sh to create samples with some single images in "single_neg" and "single_pos" folders
  5. run merge_neg_samples.sh and merge_pos_samples.sh to merge .lst files for the generated samples to the "bg.txt" and "info.dat" files
  6. copy "info/*.jpg" to "pos" folder and bg/*.jpg to "neg" folder
  7. run create_samples.sh to run opencv_createsamples command for generating .vec file
  8. run nohup ./train.sh & to run the training in background

AAAAH YEAH That is sort of complicated, You might say and yes it is.
The concept is that you have some negative and positive images in two different folders, about 1000 images in each. However, you need to generate more images with different aspects (angles, perspectives, ...).
So what we've done, is creating information data file of original images.
Afterward, we generate more than 10000 of images with different perspective and angles from a single image.
Then, we tried to make another information data file for these new images.
Finally we merged all the information files to pass through the opencv_traincascade.

If you are messed up till now, I suggest you to read following tutorials:

After training the cascade classifier, -whether with HAAR or LBP algorithms- an XML file will be generated which is needed for the final step:

Facial Expression Classification

Train CNN Data

You can find the code in CNN folder of the repository. We've used Keras library to train and predict.

The procedure of the CNN training is divided into the followong steps:

  1. Dataset should be resized into 48x48 pixels.
  2. Dataset should be divided into three section including train, validation, and test.
  3. We should generate images with different angles and scales and shifts which will be used as training and validation data.
  4. We use sequential model which contains 8 convoloutional layers and 5 dense layers.
  5. We use Dropout, Batch Normalization, and MaxPool2D layers between convoloutional and dense layers.
  6. Finally, after the training, we test our trained model with the test dataset.

You can observe the CNN model configuration in the following code snippet.

model = Sequential()
model.add(Conv2D(64, kernel_size = (3, 3), input_shape = (48, 48, 1), activation = 'relu', kernel_regularizer=l2(0.01)))
model.add(Conv2D(64, kernel_size = (3, 3), activation = 'relu', padding='same'))
model.add(BatchNormalization())
model.add(MaxPool2D(pool_size = (2, 2)))
model.add(Dropout(0.3))

model.add(Conv2D(128, kernel_size = (3, 3), activation = 'relu', padding='same'))
model.add(BatchNormalization())
model.add(Conv2D(128, kernel_size = (3, 3), activation = 'relu', padding='same'))
model.add(BatchNormalization())
model.add(MaxPool2D(pool_size = (2, 2)))
model.add((Dropout(0.4)))

model.add(Conv2D(256, kernel_size = (3, 3), activation = 'relu', padding='same'))
model.add(BatchNormalization())
model.add(Conv2D(256, kernel_size = (3, 3), activation = 'relu', padding='same'))
model.add(BatchNormalization())
model.add(MaxPool2D(pool_size = (2, 2)))
model.add((Dropout(0.5)))

model.add(Conv2D(512, kernel_size = (3, 3), activation = 'relu', padding='same'))
model.add(BatchNormalization())
model.add(Conv2D(512, kernel_size = (3, 3), activation = 'relu', padding='same'))
model.add(BatchNormalization())
model.add(MaxPool2D(pool_size = (2, 2)))
model.add((Dropout(0.5)))

model.add(Flatten())
model.add(Dense(512, activation = 'relu'))
model.add(Dropout(0.5))

model.add(Dense(256, activation = 'relu'))
model.add(Dropout(0.4))
model.add(BatchNormalization())

model.add(Dense(128, activation = 'relu'))
model.add(Dropout(0.3))

model.add(Dense(64, activation = 'relu'))
model.add(Dropout(0.3))
model.add(BatchNormalization())

model.add(Dense(5, activation = 'softmax'))
model.compile(loss = 'categorical_crossentropy', optimizer = 'adam', metrics = ['accuracy'])
model.fit_generator(generator=train_generator,
                    steps_per_epoch=4096,
                    validation_data=valid_generator,
                    validation_steps=1024,
                    epochs=1000,
                    callbacks= [model_check, earlystopping, reduce_lr]
)

The validation accuracy is 0.6767.

The validation loss is 0.8386.

The training accuracy is 0.7165.

The training accuracy is 0.7616.

Merge

This stage is to use the cascade.xml and the CNN trained model to detect the face position and the facial expression respectively.

No complexity >> We didn't push the merged codes

Results (By Kamran Akbar)

results-kamran-akbar

Authors:

  • Alireza Kavian
  • Kamran Akbar