This repository contains all the necessary scripts used for data preparation, model training and building the GUI App.
- Tensorflow 2.2.0
- numpy
- opencv
- PyQt5
- scikit-learn
Link to our npy files
The repository has four directories:
-
DataPreparation Script : Contains CreateNPY_Files.py script for making npy arrays out of the images
-
Model Training Script :
- Model.py : Contains function for model preparation
- Ensembling.py : Contains functions for ensembling and function for measuring the performance of the ensembler
- TrainModel.py : Script for training the models
- Performance : Script for getting the performance metrics
-
GUI Application : This directory contains Desktop Application made using Qt which uses the trained models and ensembling for COVID-19 detection.
- SavedModels : Contains the trained models
- app.py : Main Python application
- utils.py : Contains utility functions that is used by the application.
-
K-Fold Cross validation : Contains the notebook for performing 5-fold cross validation & scripts to get the performance metrics like Accuracy, Sensitivity, Specificity, F1-Score & AUC
1. The Image data are kept into separate directories as COVID_19 +ve and COVID_19 -ve.
2. These images are split into separate diretories as train and test
Thus the directory structure is as:
Images
|
----Train
| |
| ---- COVID_19 +ve
| |
| ---- COVID_19 -ve
|
-----Test
|
---- COVID_19 +ve
|
---- COVID_19 -ve
3. Then run the CreateNPY_Files.py script and the enter the paths according to the prompt.
1. Run the TrainModel.py script and the enter the paths.
2. The models will be saved in the directory.
1. Run Performance.py to get the performance of the ensembler and enter the paths.
2. And the accuracy, confusion matrix will be printed in the console
1. Run the app.py script