Can anyone recommend me a very good pre-trained model for face or head detection?
-
16-12-2020 - |
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.).
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.