Skip to content

This project aims to detect COVID-19 cases from chest X-ray images using the Swin Transformer model.

License

Notifications You must be signed in to change notification settings

MahtabRanjbar/covid-detection-Swin-Transormers

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

COVID-19 Detection using Swin Transformers

This project aims to detect COVID-19 cases from chest X-ray images using the Swin Transformer model. The model is trained on a dataset of chest X-ray images obtained from the Chest X-ray Dataset on Kaggle. By leveraging the power of Swin Transformers, the model achieves accurate classification of X-ray images into two categories: COVID-19 positive or normal.


Table of Contents


Installation

  1. Clone this repository to your local machine:

    git clone https://github.com/MahtabRanjbar/covid-detection-Swin-Transormers.git
  2. Navigate to the project directory:

    cd covid-detection-Swin-Transormers 
    
  3. Install the required packages:

    pip install -r requirements.txt

Usage

Usage To run the project, follow these steps:

  1. Set the necessary configurations in the config.py file. Update the DataDir variable to the directory where the dataset is located.

  2. Run the main script:

    python src/main.py
  3. The script will load and preprocess the image data, split it into training and testing sets, and create data generators for training, validation, and testing.

  4. The Swin Transformer model will be built and trained using the training data. The model will be evaluated using the testing data, and the performance metrics will be displayed, including a confusion matrix and classification report.


Dataset

The dataset used for training and testing the model is the Chest X-ray Dataset from Kaggle. It contains a collection of chest X-ray images in two categories: COVID-19 positive and normal. To use this dataset, download the dataset from provided link.

The dataset has been split into three subsets: train set, validation set, and test set.


Model

The Swin Transformer model is a state-of-the-art architecture that combines the strengths of transformers and convolutional neural networks (CNNs) for image classification tasks. It introduces a hierarchical architecture that divides the image into patches and processes them with self-attention mechanisms. This allows the model to capture global and local dependencies effectively, enabling robust feature representation.

The Swin Transformer model used in this project is implemented using TensorFlow and Keras. It consists of several layers, including patch embedding, multi-scale self-attention, and feed-forward networks.


Evaluation Metrics

After training and testing the brain tumor detection model, several evaluation metrics have been computed to assess its performance. These evaluation metrics provide insights into the model's effectiveness in detecting covid-19. The metrics include accuracy, F1 score, recall, and precision.

Metric Value
Accuracy 0.971478718
F1 Score 0.971472654
Recall 0.971478718
Precision 0.971485011

For further evaluation metrics and details, please check the reports folder


Contributing

Contributions are always welcome! If you have any ideas or suggestions, please feel free to open an issue or a pull request.


License

This project is licensed under the MIT License. See the LICENSE file for more information


Contact

If you have any questions or comments about this project, please feel free to contact me at [email protected]

About

This project aims to detect COVID-19 cases from chest X-ray images using the Swin Transformer model.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages