git clone --recursive https://github.com/arpanmangal/CovidAID.git
The main requirements are listed below
- Python 3.6.10
- PyTorch 0.3.0
- Torchvision 0.2.0
- Numpy
- Pillow 6.1
- Scikit-learn
- Pandas
- Seaborn
- Matplotlib
- Tqdm
CovidAID
uses the covid-chestxray-dataset for COVID-19 X-Ray images and chest-xray-pneumonia dataset for data on Pneumonia and Normal lung X-Ray images.
Download and save the chest-xray-pneumonia dataset in the current folder.