This projects aims to detect abnormalities in chest X-rays using methods in Deep Learning.
- Create a Dataset parser
- Visualize images
- Explore class imbalance
- Process labels (one-hot encoding, reassigning labels, etc.)
- Binary classification
- Create generators for train and validation splits
- Train using a VGG-style neural network
- Use transfer learning (InceptionResNetv2)
- Plot training graphs (Accuracy, Loss)
- Multilabel classification
- Create generators for train and validation splits
- Without sampling: Train using a MobileNet pretrained on ImageNet
- With sampling: Train using a MobileNet pretrained on ImageNet
- Plot training graphs (Accuracy, Loss, ROC-AUC)
The notebook is written in Python3 and uses the following major libraries/frameworks:
- Deep Learning framework: Keras
- Data processing: Numpy, Pandas, Scikit-learn
- Data visualization: Matplotlib, OpenCV
The data has been taken from the NIH Chest X-ray Dataset on Kaggle.