Skip to content

COVID-19 X-ray classification using PyTorch, ResNet-18, and transfer learning

Notifications You must be signed in to change notification settings

khaledjamal/detecting-covid-xray-pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

2 Commits
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

Detecting COVID-19 with Chest X-Ray using PyTorch

Repository: detecting-covid-xray-pytorch

πŸ“Œ Project Overview

This project demonstrates a complete machine learning workflow for classifying chest X-ray images into three categories: Normal, Viral Pneumonia, and COVID-19. It was developed as part of a guided hands-on Coursera project using PyTorch and ResNet-18, with a focus on medical imaging and deep learning.

πŸš€ Key Features

  • Custom Dataset and DataLoader: Created using PyTorch to manage image data effectively.
  • Data Transformations: Applied standard preprocessing techniques like resizing, normalization, and augmentation.
  • Model Training: Fine-tuned a pre-trained ResNet-18 model on a real-world chest X-ray dataset.
  • Model Evaluation: Visualized predictions, tracked training and validation performance, and used accuracy as a metric.

🧠 Skills Demonstrated

  • Image classification using Convolutional Neural Networks (CNNs)
  • Deep learning with PyTorch
  • Data augmentation and normalization
  • Custom PyTorch Dataset and DataLoader
  • Transfer learning with ResNet-18
  • Working with medical imaging datasets

🧾 Technologies Used

  • Python 3
  • NumPy
  • PyTorch
  • Torchvision
  • PIL (Python Imaging Library)
  • Matplotlib

πŸ“ Project Structure

β”œβ”€β”€ complete_notebook.ipynb        # Main Jupyter notebook with code and output
β”œβ”€β”€ complete_notebook.pdf          # PDF export for easy viewing
β”œβ”€β”€ README.md                      # Project overview and documentation

πŸ“Œ How to Use

  1. Clone this repository:
    git clone https://github.com/yourusername/detecting-covid-xray-pytorch.git
  2. Open complete_notebook.ipynb in Jupyter Notebook or VS Code.
  3. Run the cells sequentially to see preprocessing, model training, and evaluation.
  4. Alternatively, view the complete_notebook.pdf for a formatted read-only version.

πŸ“ž Contact

Khaled Jamal
Machine Learning Enthusiast & Senior Software Engineer
πŸ“§ [javaobjects@gmail.com]
πŸ”— LinkedIn Profile
🌍 Dubai, UAE


This project reflects my interest in applying machine learning to healthcare and my commitment to transitioning into AI-focused roles.

About

COVID-19 X-ray classification using PyTorch, ResNet-18, and transfer learning

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published