Skip to content

harshaparida/PneumoDetect

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PneumoDetect

Fastai PyTorch Python Psutil

stack

Project Overview

This project is aimed at developing a deep learning model to detect pneumonia from chest X-ray images using the Fastai library. The project uses a convolutional neural network (CNN) based on the ResNet34 architecture and leverages data augmentation techniques and transfer learning for improved performance.

Dataset

You can download the dataset from Kaggle: https://www.kaggle.com/paultimothymooney/chest-xray-pneumonia/download I have trained and saved the model: "model.pkl"

Requirements

Ensure you have the following dependencies installed:

  • fastai
  • torch
  • pandas
  • numpy
  • pathlib
  • psutil

You can install these dependencies using pip:

pip install fastai torch pandas numpy pathlib psutil

License

This project is licensed under the MIT License (https://github.com/harshaparida/PneumoDetect/blob/main/LICENSE).

About

This project aims to develop a deep learning model using the Fastai library to detect pneumonia from chest X-ray images. The model utilizes the ResNet34 architecture, applying data augmentation and transfer learning for improved accuracy. Steps include setting up the environment, data preparation, model training, fine-tuning, and making predictions

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors