- Notice: This project is based on PyTorch. And tensorflow version is here.
- torch == 1.3.1 (other versions would be Okay)
- numpy
- matplotlib
- opencv-python
- scipy
- trainNet1.py is the implementations of training PRNet model, and model will be saved in model/.
- landmarks detector tool should be downloaded and put into Data/face_detector/.
- Generate training data:
- Same as tensorflow version.
NOTICE: Be careful of all the PATHS used in these files, you MUST modify them by yourself.
Thanks for these contributers and their excellent works: