Skip to content

Multi-class classification of chest X-ray images using transfer learning with state-of-the-art CNNs. EfficientNetV2-S achieved 92.79% accuracy on the CoronaHack dataset.

Notifications You must be signed in to change notification settings

Felipecastanog/Final_Project_Data622

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

79 Commits
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

DATA-662 - MDCH-615

🧠 Final Project β€” Data 622

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.


πŸ“‚ Repository Structure

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)

About

Multi-class classification of chest X-ray images using transfer learning with state-of-the-art CNNs. EfficientNetV2-S achieved 92.79% accuracy on the CoronaHack dataset.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 4

  •  
  •  
  •  
  •