Machine Learning Classification Using Transfer Learning and X-Ray Images
This project explores the use of transfer learning for multi-class classification of chest X-ray images into three clinically meaningful categories: Normal, Viral Pneumonia, and Bacterial Pneumonia. Leveraging the publicly available CoronaHack Chest X-Ray Dataset, we benchmarked various state-of-the-art CNN architectures and fine-tuned them for medical image classification tasks.
π Best model: EfficientNetV2-S, achieving 92.79% test accuracy and 0.9780 micro-AUC.
Final_Project_Data622/
β
βββ Dataset/ # Metadata and label files (images downloaded separately from Kaggle)
βββ Final Code/ # Final version of training and classification scripts using EfficientNetV2-S
βββ Testing Models/ # Experiments with multiple pre-trained CNN architectures
β βββ ResNet101/
β βββ Dense121/
β βββ Xception/
β βββ ... (etc.)
β
βββ README.md # Main project README (this file)