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}
}