A Symmetric Semantic-Aware Transformer Network for Makeup Transfer and Removal

This is the official pytorch code for SSAT: A Symmetric Semantic-Aware Transformer Network for Makeup Transfer and Removal, which has been accepted by AAAI-2022. Note that only trained models and test code are provided in pytorch code. We will provide complete training code in Mindspore code in the future.

Example

We have provided test samples and trained models, you only need to run the “test.py” file and the results will be in “./results” folder .

How to run

  1. Prepare face parsing. Face parsing is used in this code. In our experiment, face parsing is generated by github.com/zllrunning/face-parsing.PyTorch.
  2. Put the results of face parsing in the .testseg1makeup and testseg1non-makeup
  3. python test.py.

Our results

.

Read more here: Source link