In this project, I have implemented a Face landmarks prediction model with fastai v1. The model was trained with a novel loss function "Wing loss" proposed in (https://arxiv.org/abs/1711.06753) this paper.
To test the model,
- Clone the repo.
- Open the notebook.
- Run the import statements in the start of the notebook.
- Run the cell with wing loss definition.
- Scroll to the bottom of the page and run the last section.