GDGAN.pytorch

A PyTorch implementation of https://arxiv.org/abs/1812.01690

Dependencies

• PyTorch 1.0.0 or later
• numpy
• scipy
• sklearn
• tensorboardX
  • see the link for more dependencies

Data

Please download NIH Chest X-ray dataset from here

Train

GDGAN

$ python main.py

See possible arguments by $ python main.py --help

Classifier

$ python classifier/classifier.py

See possible arguments by $ python classifier/classifier.py --help

Loss (hyperparameters detail)