Skip to content

seitalab/GDGAN.pytorch

Repository files navigation

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)

Releases

No releases published

Packages

No packages published

Languages