FACE RECOGNITION USING TRANSFER LEARNING

FACE RECOGNITION USING TRANSFER LEARNING

In today's AI world face recognition is becoming a great technology and everyone is using that.

Here I have used the VGG16 architecture of CNN and the pre-trained weights of imagenet Dataset.

In Transfer Learning we retain our pre-trained model but not completely, we freeze our pre-trained model after that add some more layers and then train only these layers for those new faces.

GitHub link:-

https://github.com/Atulraj-1103/face_recognition.git


Training the model

No alt text provided for this image
No alt text provided for this image

new layers are added

No alt text provided for this image

It will also save the model as "face_actor_VGG_TL.h5"

No alt text provided for this image

Now Loading the data and predicting from it

No alt text provided for this image

Now to check the model or to see the output give the image path.

No alt text provided for this image
No alt text provided for this image

The image data-set I have downloaded from Kaggle Dataset.


Thank You...!!!








To view or add a comment, sign in

More articles by ATUL RAJ

Explore content categories