Third party libraries requirements:
- numpy
- matplotlib
- Pytorch
- pandas
- scikit-learn
Please install them before running the script.
pip install <required-lib>
or using Conda
conda install <required-lib>
Download the NIH Chest X-ray dataset and extract to the images folder, the link is below: https://nihcc.app.box.com/v/ChestXray-NIHCC
And run the script with Python 3.
Run instance:
Preprocessing the labels and produce train, validation and test sets
python preprocessing.py
Train the model and the best iteration would be saved
python train.py
Test the best model with the test dataset
python test.py
It will take sometime to finish running depends on your computer specification. After the script finished running. Resuls will appear in the terminal.
Thank you for your effort reviewing my project, have a good day!
Reference Zoogzog/Chexnet Arnoweng/CheXNet