tFace Detection by Transfer Learning by using resnet model

Gaurav Khore
2 min readMay 19, 2020

--

Resnet is a per trained model on the imagenet dataset and we can use weights of this model to train our own model and from this model we can predict the face.

For the dataset i have use the harrcascade classifier. For this i use cv2 library to click images from the webcam of my laptop and by harrcascade i detect the face and crop the image accordingly . By doing this my dataset get a proper collection of images.

Link of code for collecting dataset:- https://github.com/Gaurav-Khore/Face_detection/blob/master/pic_capture.ipynb

Then i used resnet pretrained model weights to predict the data i collected. Also add 1 Flatten layer , 3 dense layer into this pre trained model and done transfer learning only on this layers by using weights from resnet.

Link of code for Transfer learning and Face Detection:- https://github.com/Gaurav-Khore/Face_detection/blob/master/faceDection_resnet.ipynb

--

--