---
license: apache-2.0
datasets:
- Arunisto/brain_tumor_dataset
language:
- en
base_model:
- microsoft/swin-tiny-patch4-window7-224
pipeline_tag: image-classification
library_name: transformers
tags:
- computer_vision
- image_classification
---

# Pre-trained swin transformer model for brain tumor classification

## Model Description
This is pre-trained swintransformer model for classify brain tumor.

## Training Dataset
This model trained with MRI Scan images of brain  tumor `healthy` and `tumor`

### Dataset Format
- **Images:** JPEG/PNG formats
- **Dataset:** Arunisto/brain_tumor_dataset

## Training Details
- **Model:** microsoft/swin-tiny-patch4-window7-224
- **Framework:** Transformers
- **Epochs:** 10


## Usage
This model can be used to detect two type of brain condition is `healthy` or `tumor`


## Limitations
- The model is perform well only with brain top-side only, and for healthy it's only predicted with the score of 65% only 

## Future Work
- needs to run more epochs
- trying to classify different tumor conditions
  
## License
Apache 2.0

## Citation
If you use this model in your research or projects, please cite it as follows:

```
@misc{arun-arunisto2024yolov8_bike_odometer,
  title={Pre-trained swin transformer model for brain tumor classification},
  author={Arun Arunisto},
  year={2024},
  howpublished={\url{https://huggingface.co/Arunisto/brain_tumor_classification}},
}
```