├── .github └── workflows │ └── python-publish.yml ├── .gitignore ├── LICENSE ├── README.md ├── distributed_train.sh ├── eval.sh ├── figures ├── Compare.png └── Top1.jpg ├── flops_computation.py ├── generate_label.py ├── main.py ├── real.json ├── requirements.txt ├── seg ├── configs │ └── lvvit │ │ ├── upernet_lvvit_l_512x512_160k_ade20k.py │ │ ├── upernet_lvvit_m_512x512_160k_ade20k.py │ │ └── upernet_lvvit_s_512x512_160k_ade20k.py └── mmseg │ └── models │ └── backbones │ └── vit.py ├── setup.py ├── tlt ├── __init__.py ├── data │ ├── __init__.py │ ├── dataset.py │ ├── label_transforms_factory.py │ ├── loader.py │ ├── mixup.py │ └── random_augment_label.py ├── loss │ ├── __init__.py │ └── cross_entropy.py ├── models │ ├── __init__.py │ ├── layers.py │ └── lvvit.py └── utils │ ├── __init__.py │ └── utils.py ├── validate.py └── visualize ├── baselines └── ViT │ └── LVViT_LRP.py └── lvvit_example.ipynb /.github/workflows/python-publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/.github/workflows/python-publish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/README.md -------------------------------------------------------------------------------- /distributed_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/distributed_train.sh -------------------------------------------------------------------------------- /eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/eval.sh -------------------------------------------------------------------------------- /figures/Compare.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/figures/Compare.png -------------------------------------------------------------------------------- /figures/Top1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/figures/Top1.jpg -------------------------------------------------------------------------------- /flops_computation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/flops_computation.py -------------------------------------------------------------------------------- /generate_label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/generate_label.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/main.py -------------------------------------------------------------------------------- /real.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/real.json -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | timm==0.4.5 2 | scipy -------------------------------------------------------------------------------- /seg/configs/lvvit/upernet_lvvit_l_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/seg/configs/lvvit/upernet_lvvit_l_512x512_160k_ade20k.py -------------------------------------------------------------------------------- /seg/configs/lvvit/upernet_lvvit_m_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/seg/configs/lvvit/upernet_lvvit_m_512x512_160k_ade20k.py -------------------------------------------------------------------------------- /seg/configs/lvvit/upernet_lvvit_s_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/seg/configs/lvvit/upernet_lvvit_s_512x512_160k_ade20k.py -------------------------------------------------------------------------------- /seg/mmseg/models/backbones/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/seg/mmseg/models/backbones/vit.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/setup.py -------------------------------------------------------------------------------- /tlt/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /tlt/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/tlt/data/__init__.py -------------------------------------------------------------------------------- /tlt/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/tlt/data/dataset.py -------------------------------------------------------------------------------- /tlt/data/label_transforms_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/tlt/data/label_transforms_factory.py -------------------------------------------------------------------------------- /tlt/data/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/tlt/data/loader.py -------------------------------------------------------------------------------- /tlt/data/mixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/tlt/data/mixup.py -------------------------------------------------------------------------------- /tlt/data/random_augment_label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/tlt/data/random_augment_label.py -------------------------------------------------------------------------------- /tlt/loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/tlt/loss/__init__.py -------------------------------------------------------------------------------- /tlt/loss/cross_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/tlt/loss/cross_entropy.py -------------------------------------------------------------------------------- /tlt/models/__init__.py: -------------------------------------------------------------------------------- 1 | from .lvvit import * 2 | -------------------------------------------------------------------------------- /tlt/models/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/tlt/models/layers.py -------------------------------------------------------------------------------- /tlt/models/lvvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/tlt/models/lvvit.py -------------------------------------------------------------------------------- /tlt/utils/__init__.py: -------------------------------------------------------------------------------- 1 | from .utils import load_pretrained_weights -------------------------------------------------------------------------------- /tlt/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/tlt/utils/utils.py -------------------------------------------------------------------------------- /validate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/validate.py -------------------------------------------------------------------------------- /visualize/baselines/ViT/LVViT_LRP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/visualize/baselines/ViT/LVViT_LRP.py -------------------------------------------------------------------------------- /visualize/lvvit_example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zihangJiang/TokenLabeling/HEAD/visualize/lvvit_example.ipynb --------------------------------------------------------------------------------