This repository includes all of the necessary files related to the second DS 4002 project for our group, The Swifties. This README.md file contains an overview of what is included in each of the three folders: SRC, DATA, and FIGURES. The LICENSE.md file explains to a visitor the terms under which they may use and cite this repository. The SRC folder contains all source code for the project. The DATA folder contains all of the data for the project. Lastly, the FIGURES folder contains all figures generated over the course of the project.

We pulled the source data from Kaggle. It includes images of chest X-rays of patients with COVID-19, patients with viral pnemonia, and healthy lungs. The dataset consists of six different directories, split by recommending testing and training data, each containing a directory for each diagnosed condition.

We downloaded the data as a collection of image files and cleaned it for use in Python. Make sure the data is available and accessible in the same working directory as the file, and make any neccessary changes to the file path in order for the dataset to be available on your device.

This code can be used by running py .\example.py using Python version 3.11.2.

Our dataset can be found here [1].

It is important to note that the dataset only contains pictures from those affected by COVID-19, Viral Pneumonia, or those unaffected by disease, not people affected by other diseases involving the lungs. This is what caused us to limit our model to these three categories.

[1] Raikote, Pranav, “Covid-19 Image Dataset,” Kaggle.com, CC BY-SA 4.0, 2020. [Online]. Available: https://www.kaggle.com/datasets/pranavraikokte/covid19-image-dataset. [Accessed March 14, 2023].

Our submission for M1 can be found here.

Our submission for M2 is linked here.