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}},
}
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social
visibility and check back later, or deploy to Inference Endpoints (dedicated)
instead.
Model tree for Arunisto/brain_tumor_classification
Base model
microsoft/swin-tiny-patch4-window7-224