COVID-19 X-Ray Detection
Detecting COVID-19, Pneumonia and Healthy X-Rays using PyTorch and Wandb
The github repository for this project can be found here.
The goal of this project was to detect if a person has COVID-19, Pneumonia or if they are healthy by looking at their X-Ray images. A pretrained ResNet18 model was used for this task. The dataset was provided by Mendely Data and can be found here.
PyTorch was used for model creation and traning. The model was done locally and wandb was used for tracking the model training process and visualizing the results. An example can be found here.
How the probabilities look like in wandb