1. extract_faces.ipynb
- data cleaning2. pytorch.ipynb
- training, validation3. eval.ipynb
- testing, visualization of results
main.py
- entrypoint for running the live facial feature detection (%run -i main.py
in ipython console)
To run main.py
or webpage.py
, you need the model files. You can either:
- download the pretrained models here:
- cls.pth (required)
- cae.pth (optional)
- drag and drop the models in
resources/models/
OR
- download the CelebA dataset and train your own models by running the
notebooks
in order