Skip to content

jdelanoy/STGAN-pytorch

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Requirements

Organization of the code

  • agents : The main training files (architecture, optimisation scheme) are in agents. trainingModule is an abstract agent that regroup some basic training functions (similar to Pytorch lighting abstraction. Most of the optimization/training procedure is is faderNet (the most simple one), the others build on it by just changing a few key functions (group the inputs, define the nets):
    • faderNet : architecture of G1 without the normals, contains most of the code (equivalent to faderNet)
    • faderNet_with_normals : G1
    • faderNet_with_normals_2steps : G2
    • faderNet_with_normals_illum : a test of G1 + illumination map
  • configs : configuration files to launch the trainings or test
  • datasets : code to read the datasets
  • experiments : snapshots of experiments
  • models : code of the networks
  • utils : various utilities

Training

About

STGAN: A Unified Selective Transfer Network for Arbitrary Image Attribute Editing

Resources

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 100.0%