Thanks to visit codestin.com
Credit goes to github.com

Skip to content

sparsh0106/X-Ray-Lung-Classifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

4 Commits
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

Pneumonia Detection frmo Chest X-Rays 🫁

THis repository presents a deep learning pipeline for binary classification of chest X-Ray images into Normal and Pneumonia classes using a fine-tuned ResNet-50 model.

🧠 Project Objective

Develop an end-to-end deep learning model to identify pneumonia in chest X-rays, assisting medical professionals in rapid and reliable diagnosis.

πŸ“‚ Repository Structure

  • main.ipynb: Full pipeline for loading data, training, evaluating, and visualizing the model.
  • resnet_50.ipynb: Implementation and fine-tuning of the ResNet-50 architecture.

πŸ”’ Dataset

  • Total Images: 5,21 chset X-ray images
  • Classes: Pneumonia and Normal

Dataset is split into 'train', 'val' and 'test' directories.

πŸ”§ Features

  • 🧠 Model: ResNet-50 with pretrained ImageNet weights
  • πŸ§ͺ Task: Binary Classification - Normal vs Pneumonia
  • πŸŒ€ Transforms: Resize, Normalize, Augmentations using torchvision.transforms
  • πŸ’₯ Loss Function: CrossEntropyLoss
  • πŸš€ Optimizer: Adam or SGD with scheduler
  • πŸ“Š Metrics: Accuracy, Precision, Recall, F1-score
  • πŸ“‰ Visualizations: Training & validation curves using Matplotlib

πŸ› οΈ Tech Stack

  • Python 3
  • PyTorch
  • Torchvision
  • Matplotlib
  • NumPy
  • scikit-learn

πŸ“Ž Notes

  • THis project uses transfer learning - leveraging a ResNet-50 pretrained on ImageNet
  • Ideal for binary medical image classification tasks
  • GPU support is recommended for faster training.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published