Skip to content

Implementation of Disentangled Makeup Transfer with Generative Adversarial Network

Notifications You must be signed in to change notification settings

yan86471/DMT-implementation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

DMT

Introduction

A implementation of Disentangled Makeup Transfer with Generative Adversarial Network.


Training

  1. Download MT (Makeup Transfer) dataset from here.

  2. Put MT (Makeup Transfer) dataset to .\data\RawData. Your data path will like this:

.\data\RawData\images\makeup\*.png
.\data\RawData\images\non-makeup\*.png

.\data\RawData\segs\makeup\*.png
.\data\RawData\segs\non-makeup\*.png
  1. run python convert.py

  2. Modify train.py and start training.
    python train.py

  3. run python export.py and you will get h5 model in .\Export.

Demo

  1. make sure you have run python export.py to get h5 model.

  2. Modify demo.py and run python demo.py, you will find the transfer result in .\Transfer.

Some issues to know

  1. The test environment is

    • Python 3.7
    • tensorflow-gpu 2.0.0
    • tensorflow-addons 0.7.1
    • imgaug 0.4.0
  2. This is still not a completed implementation, but almost 95% is the same as paper described.

About

Implementation of Disentangled Makeup Transfer with Generative Adversarial Network

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages