MouxiaoHuang / myPRNet-PyTorch

Personal implementation of PRNet based on PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Simple implementation of PRNet (torch version)

  • Notice: This project is based on PyTorch. And tensorflow version is here.

1. Environment required
  • torch == 1.3.1 (other versions would be Okay)
  • numpy
  • matplotlib
  • opencv-python
  • scipy
2. Introductions of some files
  • 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:

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:

About

Personal implementation of PRNet based on PyTorch


Languages

Language:Python 100.0%