← all repositories

zihangJiang/TokenLabeling

PyTorch implementation of LV-ViT (Large-scale Vision Transformer) with token labeling for improved image classification and segmentation.

433 stars Jupyter Notebook Computer VisionML Frameworks
TokenLabeling
Velocity · 7d
+0.2
★ / day
Trend
steady
star history

This repository implements the paper ‘All Tokens Matter: Token Labeling for Training Better Vision Transformers’. It provides training code and pre-trained models for LV-ViT, a vision transformer architecture trained using a token labeling approach that assigns labels to every token rather than just the CLS token. The implementation is based on the timm (pytorch-image-models) library and includes scripts for label data generation and a segmentation model variant.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.