Skip to content

Implementation of PFLD A Practical Facial Landmark Detector by Tensorflow 2.

Notifications You must be signed in to change notification settings

notplus/PFLD-TF2

Repository files navigation

PFLD Tensorflow 2

Implementation of PFLD A Practical Facial Landmark Detector by Tensorflow 2.

Usage

  1. Requirements: tensorflow >= 2.0.0, numpy, opencv, pytorch (optional)

  2. Datasets

    WFLW Dataset Download

    Wider Facial Landmarks in-the-wild (WFLW) is a new proposed face dataset. It contains 10000 faces (7500 for training and 2500 for testing) with 98 fully manual annotated landmarks.

    WFLW Training and Testing images Google Drive Baidu Drive

    WFLW Face Annotations

    Unzip above two packages and put them on ./data/WFLW/

    move Mirror98.txt to WFLW/WFLW_annotations

    $ cd data

    $ python3 SetPreparation.py

    Generate tfrecord:

    python tools/generate_tfrecord.py
  3. Train

    You can change configurations in train.py and config.py. For training, just execute one line code.

    python train.py
  4. Test

    Just read test.py and load weight you want.

  5. Camera test

    You should check if pytorch is installed, and load weight you want.

Others

  1. For loss function, attributes_w_n may all be zero, which makes loss equal to zero. So it may need to rethink about the weight.

  2. We provide a model called PFLD_wing_loss_fn which uses wing_loss and removes auxiliarynet.

  3. A model called PFLD_Ultralight is available, which uses GhostBottleneck.

Reference

  1. PFLD-pytorch
  2. PFLD-Ultralight

About

Implementation of PFLD A Practical Facial Landmark Detector by Tensorflow 2.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages