State-of-the-art 68-point facial landmark detection using High-Resolution Networks (HRNet).
- 🚀 High-Resolution Networks: Maintains high resolution throughout the process
- 🎯 68 Facial Landmarks: Precise detection of key facial points
- ⚡ Real-time Performance: Optimized for live video processing
- 🔥 Multi-Scale Fusion: Advanced feature fusion across different resolutions
- 📊 Low NME Score: Superior accuracy on 300W dataset
Datasets: 300W
- Download the datasets from official sources.
- Download the annotations files from OneDrive.
To train the model, run:
- Configure your dataset path in utils/config.py for training
$ python main.py --trainFor testing the model, use:
$ python main.py --testTo run the real-time facial landmark detection:
$ python main.py --demo| NME | NME | pretrained model | model |
|---|---|---|---|
| HRNetV2-W18 | 3.3 | ImageNet | best.pt |