├── .gitignore ├── LICENSE ├── README.md ├── config ├── finetune │ ├── vit_base_finetune.py │ ├── vit_small_finetune.py │ └── vit_tiny_finetune.py ├── knn │ └── knn.py ├── linear │ ├── vit_base_linear.py │ ├── vit_small_linear.py │ └── vit_tiny_linear.py └── pretrain │ ├── vit_base_pretrain.py │ ├── vit_small_pretrain.py │ └── vit_tiny_pretrain.py ├── images ├── framework.png └── visualization.png ├── main_finetune.py ├── main_knn.py ├── main_linear.py ├── main_pretrain.py ├── module ├── augmentation.py ├── frame │ ├── contrast_momentum.py │ └── contrast_no_momentum.py ├── loss.py ├── mix.py └── vits.py ├── requirements.txt └── utils ├── logger.py └── misc.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/README.md -------------------------------------------------------------------------------- /config/finetune/vit_base_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/config/finetune/vit_base_finetune.py -------------------------------------------------------------------------------- /config/finetune/vit_small_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/config/finetune/vit_small_finetune.py -------------------------------------------------------------------------------- /config/finetune/vit_tiny_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/config/finetune/vit_tiny_finetune.py -------------------------------------------------------------------------------- /config/knn/knn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/config/knn/knn.py -------------------------------------------------------------------------------- /config/linear/vit_base_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/config/linear/vit_base_linear.py -------------------------------------------------------------------------------- /config/linear/vit_small_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/config/linear/vit_small_linear.py -------------------------------------------------------------------------------- /config/linear/vit_tiny_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/config/linear/vit_tiny_linear.py -------------------------------------------------------------------------------- /config/pretrain/vit_base_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/config/pretrain/vit_base_pretrain.py -------------------------------------------------------------------------------- /config/pretrain/vit_small_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/config/pretrain/vit_small_pretrain.py -------------------------------------------------------------------------------- /config/pretrain/vit_tiny_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/config/pretrain/vit_tiny_pretrain.py -------------------------------------------------------------------------------- /images/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/images/framework.png -------------------------------------------------------------------------------- /images/visualization.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/images/visualization.png -------------------------------------------------------------------------------- /main_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/main_finetune.py -------------------------------------------------------------------------------- /main_knn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/main_knn.py -------------------------------------------------------------------------------- /main_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/main_linear.py -------------------------------------------------------------------------------- /main_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/main_pretrain.py -------------------------------------------------------------------------------- /module/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/module/augmentation.py -------------------------------------------------------------------------------- /module/frame/contrast_momentum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/module/frame/contrast_momentum.py -------------------------------------------------------------------------------- /module/frame/contrast_no_momentum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/module/frame/contrast_no_momentum.py -------------------------------------------------------------------------------- /module/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/module/loss.py -------------------------------------------------------------------------------- /module/mix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/module/mix.py -------------------------------------------------------------------------------- /module/vits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/module/vits.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/requirements.txt -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/visresearch/patchmix/HEAD/utils/misc.py --------------------------------------------------------------------------------