Skip to content

A Deep learning model to predict Pneumonia and determine the location on the Chest X-Ray image

License

Notifications You must be signed in to change notification settings

Asv53/Pneumonia-classification-and-segmentation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

31 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Pneumonia-classification-and-segmentation

A Deep learning model to predict Pneumonia and determine the location on the Chest X-Ray image This project aims to classfy whether given chest X-Ray image has Pneumonia or not and if the image Pneumonia then we determine the location in the image.

SOFTWARE REQUIREMENTS

  1. Nvidia CUDA Toolkit 12.1
  2. cudnn
  3. PyTorch cuda
  4. VS Code
  5. Python

DATASETS USED

RSNA Pneumonia Challenge dataset https://www.kaggle.com/competitions/rsna-pneumonia-detection-challenge/data Chest X-Ray Images (Pneumonia) https://www.kaggle.com/datasets/paultimothymooney/chest-xray-pneumonia

REQUIREMENTS

torch torchvision matplotlib sklearn numpy pandas cv2 albumentations pydicom PIL

Extract the datasets and place in appropriate folders

Code is available in project - final.ipynb file.

About

A Deep learning model to predict Pneumonia and determine the location on the Chest X-Ray image

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published