This is a deep learning project that aims to detect facial landmarks in images using PyTorch. The model is based on the ResNet-18 architecture and is trained on the iBUG 300W Large Face Landmark Dataset.
Facial landmarks detection is an important task in computer vision and has various applications, including face recognition, emotion analysis, and facial expression synthesis. In this project, we use a deep learning model based on the ResNet-18 architecture to accurately predict 68 facial landmarks in an image.
- Python 3.x
- PyTorch
- torchvision
- NumPy
- Matplotlib
- OpenCV
To install the required libraries, run the following command:
pip install torch torchvision numpy matplotlib opencv-python
The iBUG 300W Large Face Landmark Dataset contains images of faces along with their corresponding ground-truth facial landmarks annotations. The dataset is used for both training and testing the model.
You can download the dataset from the following link: iBUG 300W Large Face Landmark Dataset
To use the model for facial landmarks detection, you can follow these steps:
- Clone this repository to your local machine.
- Download the iBUG 300W Large Face Landmark Dataset and place it in the
data
directory. - Install the required libraries using the installation instructions above.
I would like to thank the authors of the iBUG 300W Large Face Landmark Dataset for providing the valuable dataset for this project.
This project is licensed under the MIT License - see the LICENSE file for details.