kornia/yunet

Pretrained weights for YuNet, used by kornia.models.FaceDetector.

YuNet is a lightweight face detection CNN based on depthwise-separable convolutions and a multi-scale FPN head. Runs in real time on CPU. IEEE TBIOM 2021.

Original repo: ShiqiYu/libfacedetection.train

Weights

File	Description
yunet_final.pth	Default checkpoint

Citation

@article{facedetect-yu,
    author  = {Yuantao Feng and Shiqi Yu and Hanyang Peng
               and Yan-ran Li and Jianguo Zhang},
    title   = {Detect Faces Efficiently: A Survey and Evaluations},
    journal = {IEEE Transactions on Biometrics, Behavior, and Identity Science},
    year    = {2021}
}