Downloads · 30 days
5
3% of all-time downloads
dementia-vit
rmezapi/dementia-vit
dementia-vit is a image classification model from rmezapi. Use it when you need a label for an image. It is set up for transformers. The card lists the license as mit.
This project was intended to test the limits of the ViT on a tough dementia dataset. The data used can be found on HuggingFace at: https://huggingface.co/datasets/Falah/AlzheimerMRI. The project follows closely the fo…
All-time downloads
148
Public
Repo size
687 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin343 MB · 100%
At a glance
- Task
- Image Classification
- Library
- transformers
- License
- mit
- Model type
- vit
- Access
- Public
- Created
- Mar 10, 2025
- Updated
- Mar 10, 2025
- SHA
- c3e3ef70
Base models
Datasets
Model card
From the Hugging Face model README
This project was intended to test the limits of the ViT on a tough dementia dataset. The data used can be found on HuggingFace at: https://huggingface.co/datasets/Falah/Alzheimer_MRI. The project follows closely the following tutorials:
https://www.youtube.com/watch?v=r88L_yLJ4CE&ab_channel=code_your_own_AI https://www.youtube.com/watch?v=qU7wO02urYU&ab_channel=JamesBriggs I modify the code presented in the video and tune all parameters to optimize performance using mostly the same libraries and tools. This is a practice project for myself as I return to coding/designing ML models after dedicating time to AI/ML theory (model architectures, transfer learning)
- Task
- Image Classification
- Library
- transformers
- Type
- vit
- License
- mit
- Languages
- en
- Created
- Mar 10, 2025
- Updated
- Mar 10, 2025