Skip to content

Look into means of improving classification of covid-19 cases with the help of data augmentation.

Notifications You must be signed in to change notification settings

hanss0n/pneumonia_detection

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

62 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Pneumonia Detection

This project explores diagnosing pneumonia through chest x-ray images with the help of deep learning (TensorFlow & Keras) and various data augmentation methods. The dataset used for this project is part of a Kaggle challenge and can be found here.

Neural Network

The neural network used is written in Keras and consists of three convolution blocks as well as a max pool layer for each one of them. On the top there's a fully connected layer with 512 units which is activated using relu-activation. We also added Dropout layers to two of the convolution layers to reduce over fitting due to the scarcity of data.

Preprocessing

The images in the dataset were scaled to 150x150 pixels and converted from RGB to grayscale before they were fed to the model.

Data Augmentation

The data augmentation methods that were tested were Cutmix, Mixup and Cutout, as well as combinations of these.

Cutmix

Before Cutmix After Cutmix

Link to paper

Mixup

Before Mixup After Mixup

Link to paper

Cutout

Before Cutout After Cutout

Link to paper

About

Look into means of improving classification of covid-19 cases with the help of data augmentation.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages