/Tf-CoronaXray

A Deep Learning approach at classifying lung X-rays between COVID-19 affected and Normal cases.

Primary LanguageJupyter NotebookMIT LicenseMIT

Tf-CoronaXray

Model summary:

summary

3 Convolution layers accompanied by max pooling layers with flatten to compress the parameters into a column vector containing all the contents of the larger vectors. Sigmoid used for the final binary classification with ReLU as hidden layer activation function. Mini-batches of 10 images were trained and validated every epoch (Total 15 epochs). The dataset was obtained from Kaggle (dated Feb. 2020).

epochs

Model Test and validation data accuracy and loss visualised:

Accplot Lossplot

P.s: The code in both main.py and main.ipynb is the same and i've just put them both here for flexibity of usage.