This repository applies the Vision Transformer (ViT) in PyTorch to identify glomeruli in human kidney tissue images for the HuBMAP Kaggle competition.
- About
- Datasets and Weights
- Evaluation
- Score
- Usage
- Training and Inference Notebooks
- Contribution and Upvote
The Vision Transformer Segmentation project implements ViT in PyTorch for the HuBMAP Kaggle competition. The goal is to identify glomeruli in human kidney tissue images using the power of transformers in computer vision tasks.
- Training Dataset: hubmap-pytorch-vit-segmentation-train-datasets
- Inference Dataset: hubmap-pytorch-vit-segmentation-inference-datasets
This competition is evaluated on the mean Dice coefficient, which compares the pixel-wise agreement between a predicted segmentation and its corresponding ground truth.
- Private Score: 0.9354
- Public Score: 0.9064
To use this Vision Transformer Segmentation project, follow these steps:
- Clone the repository:
git clone https://github.com/elcaiseri/Vision-Transformer-Segmentation.git
- Explore the implementation and code for ViT in PyTorch.
If you find this project helpful or interesting, consider giving it a star! ⭐
© 2023 elcaiseri