Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/mahtabranjbar/covid-detection-swin-transormers
This project aims to detect COVID-19 cases from chest X-ray images using the Swin Transformer model.
https://github.com/mahtabranjbar/covid-detection-swin-transormers
computer-vision deep medica swintransformer transfor
Last synced: about 1 month ago
JSON representation
This project aims to detect COVID-19 cases from chest X-ray images using the Swin Transformer model.
- Host: GitHub
- URL: https://github.com/mahtabranjbar/covid-detection-swin-transormers
- Owner: MahtabRanjbar
- License: mit
- Created: 2023-09-16T19:47:11.000Z (over 1 year ago)
- Default Branch: main
- Last Pushed: 2023-11-22T13:11:43.000Z (about 1 year ago)
- Last Synced: 2024-01-28T17:01:32.939Z (11 months ago)
- Topics: computer-vision, deep, medica, swintransformer, transfor
- Language: Python
- Homepage:
- Size: 182 KB
- Stars: 1
- Watchers: 2
- Forks: 1
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# 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](#installation)
- [Usage](#usage)
- [Dataset](#dataset)
- [Model](#model)
- [Evaluation Metrics](#evaluation-metrics)
- [Contributing](#contributing)
- [License](#license)
- [Contact](#contact)
---
## Installation
1. Clone this repository to your local machine:
```sh
git clone https://github.com/MahtabRanjbar/covid-detection-Swin-Transormers.git
```
2. Navigate to the project directory:
```sh
cd covid-detection-Swin-Transormers
```
3. Install the required packages:
```sh
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:
```sh
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](https://www.kaggle.com/datasets/andyczhao/covidx-cxr2). 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](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](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]