Skip to content

Latest commit

 

History

History
25 lines (25 loc) · 827 Bytes

README.md

File metadata and controls

25 lines (25 loc) · 827 Bytes
Raw

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.