Purpose: To create usable Chest Xrays with a GAN as a way to augment medical data without needing to worry about privacy and accessibility.
- Create an implementation of DenseNet in PyTorch from : https://arxiv.org/abs/1608.06993
- Create an implementation of StyleGan in PyTorch from : https://github.com/NVlabs/stylegan
- Generate images from StyleGan
- Train the DenseNet two different times, with the real images and the StyleGan generated images
- Compare the scores
All the data was used can be found here: https://www.kaggle.com/nih-chest-xrays/data
Tasks:
Task | Complete |
---|---|
Gather Data | ✔️ |
Code DenseNet | ✔️ |
Code StyleGan | |
Generate images | |
Train two DenseNets | |
Analyze results |