Arunisto/brain_tumor_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}},
}