PyTorch evaluation code and pretrained models for SLPT

PyTorch evaluation code and pretrained models for SLPT (Sparse Local Patch Transformer).

Installation

Note: this released version was tested on Python3.8, Pytorch 1.10.2 and Windows 11.

Install python dependencies:

pip3 install -r requirements.txt

Run Evaluation on WFLW dataset

1. Download and process WFLW dataset

   • Download WFLW dataset and annotation from Here.
   • Unzip WFLW dataset and annotations and move files into ./dataset directory. Your directory should look like this:

     SLPT
     └───Dataset
        │
        └───WFLW
           │
           └───WFLW_annotations
           │   └───list_98pt_rect_attr_train_test
           │   │
           │   └───list_98pt_test
           └───WFLW_images
               └───0--Parade
               │
               └───...
     

      

      

      

     To finish reading, please visit source site