Question

I really need to know the best pre-trained models to detect faces and/or peoples' head. Not a face recognition model, but only to classify whether an object is a person's head/face or not.

I'm implementing one on top of Resnet50 from tensorflow.keras.applications.resnet50, but I'm not sure if it's a good approach.

Other models like centernet and efficientnet in tensorflow are pre-trained to detect several objects, but I'm not sure if I can also use it for this purpose.

I need something better than those provided by OpenCV (e.g. cv2.HOGDescriptor, cv2.CascadeClassifier, etc.).

Was it helpful?

Solution

Using facenet pretrained model to detect faces in image, but i thing ti will not detect the head, so i suggest that you train object detection model to detect faces and heads.

Licensed under: CC-BY-SA with attribution
Not affiliated with datascience.stackexchange
scroll top