Chest X-Ray Pneumonia Classification using Auxiliary Deep Generative Models
In order to run the train code you need some python modules. You can install it with pip (an anaconda enviroment is suggested) by run the command
pip install torch torchvision imgaug pydicom pandas skimage sklearn
The training dataset can be downloaded at Dataset. Once downloaded extract the data in the Kaggle/all folder.
The training can be run via jupyter notebook as well as in a python script.
- Stefano Cerri - DTU Compute
This project is licensed under the MIT License - see the LICENSE.md file for details
This repo used code from (https://github.com/wohlert/semi-supervised-pytorch) Many thanks for sharing the code!