├── .gitignore ├── LICENSE ├── README.md ├── assets └── tokenmix.png ├── data ├── train.txt └── val.txt ├── datasets.py ├── deit.py ├── engine.py ├── exp └── deit_small │ └── run.sh ├── imagenet_dataset.py ├── losses.py ├── main.py ├── relabel ├── dataset.py ├── label_transforms_factory.py ├── loader.py └── random_augment_label.py ├── samplers.py ├── tokenmix.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sense-X/TokenMix/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sense-X/TokenMix/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sense-X/TokenMix/HEAD/README.md -------------------------------------------------------------------------------- /assets/tokenmix.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sense-X/TokenMix/HEAD/assets/tokenmix.png -------------------------------------------------------------------------------- /data/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sense-X/TokenMix/HEAD/data/train.txt -------------------------------------------------------------------------------- /data/val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sense-X/TokenMix/HEAD/data/val.txt -------------------------------------------------------------------------------- /datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sense-X/TokenMix/HEAD/datasets.py -------------------------------------------------------------------------------- /deit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sense-X/TokenMix/HEAD/deit.py -------------------------------------------------------------------------------- /engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sense-X/TokenMix/HEAD/engine.py -------------------------------------------------------------------------------- /exp/deit_small/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sense-X/TokenMix/HEAD/exp/deit_small/run.sh -------------------------------------------------------------------------------- /imagenet_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sense-X/TokenMix/HEAD/imagenet_dataset.py -------------------------------------------------------------------------------- /losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sense-X/TokenMix/HEAD/losses.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sense-X/TokenMix/HEAD/main.py -------------------------------------------------------------------------------- /relabel/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sense-X/TokenMix/HEAD/relabel/dataset.py -------------------------------------------------------------------------------- /relabel/label_transforms_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sense-X/TokenMix/HEAD/relabel/label_transforms_factory.py -------------------------------------------------------------------------------- /relabel/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sense-X/TokenMix/HEAD/relabel/loader.py -------------------------------------------------------------------------------- /relabel/random_augment_label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sense-X/TokenMix/HEAD/relabel/random_augment_label.py -------------------------------------------------------------------------------- /samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sense-X/TokenMix/HEAD/samplers.py -------------------------------------------------------------------------------- /tokenmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sense-X/TokenMix/HEAD/tokenmix.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sense-X/TokenMix/HEAD/utils.py --------------------------------------------------------------------------------