/ViT-breast_cancer

This repo contains source code of an experiment involving an application of vision transformers (ViT) to health image data classification. The pretrained vision transformer, vit_b_16, is fine-tuned on a dataset of breast ultrasound scans (BUSI).

Primary LanguageJupyter Notebook

Breast Cancer Transformer

This is a personal research project that fine-tunes a pretrained vision transformer model on a breast cancer dataset. The pretrained model used in this experiment is the vit_b_16. Other vision transformer models will be explored.