├── .gitignore ├── README.md ├── figures └── framework.png ├── image_classification ├── README.md ├── datasets.py ├── engine.py ├── logger.py ├── losses.py ├── main.py ├── mcloader │ ├── __init__.py │ ├── classification.py │ ├── data_prefetcher.py │ ├── image_list.py │ ├── imagenet.py │ └── mcloader.py ├── models │ ├── __init__.py │ └── scalablevit.py ├── requirements.txt ├── samplers.py └── utils.py ├── object_detection └── README.md └── semantic_segmentation └── README.md /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yangr116/ScalableViT/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yangr116/ScalableViT/HEAD/README.md -------------------------------------------------------------------------------- /figures/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yangr116/ScalableViT/HEAD/figures/framework.png -------------------------------------------------------------------------------- /image_classification/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yangr116/ScalableViT/HEAD/image_classification/README.md -------------------------------------------------------------------------------- /image_classification/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yangr116/ScalableViT/HEAD/image_classification/datasets.py -------------------------------------------------------------------------------- /image_classification/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yangr116/ScalableViT/HEAD/image_classification/engine.py -------------------------------------------------------------------------------- /image_classification/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yangr116/ScalableViT/HEAD/image_classification/logger.py -------------------------------------------------------------------------------- /image_classification/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yangr116/ScalableViT/HEAD/image_classification/losses.py -------------------------------------------------------------------------------- /image_classification/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yangr116/ScalableViT/HEAD/image_classification/main.py -------------------------------------------------------------------------------- /image_classification/mcloader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yangr116/ScalableViT/HEAD/image_classification/mcloader/__init__.py -------------------------------------------------------------------------------- /image_classification/mcloader/classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yangr116/ScalableViT/HEAD/image_classification/mcloader/classification.py -------------------------------------------------------------------------------- /image_classification/mcloader/data_prefetcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yangr116/ScalableViT/HEAD/image_classification/mcloader/data_prefetcher.py -------------------------------------------------------------------------------- /image_classification/mcloader/image_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yangr116/ScalableViT/HEAD/image_classification/mcloader/image_list.py -------------------------------------------------------------------------------- /image_classification/mcloader/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yangr116/ScalableViT/HEAD/image_classification/mcloader/imagenet.py -------------------------------------------------------------------------------- /image_classification/mcloader/mcloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yangr116/ScalableViT/HEAD/image_classification/mcloader/mcloader.py -------------------------------------------------------------------------------- /image_classification/models/__init__.py: -------------------------------------------------------------------------------- 1 | from .scalablevit import * -------------------------------------------------------------------------------- /image_classification/models/scalablevit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yangr116/ScalableViT/HEAD/image_classification/models/scalablevit.py -------------------------------------------------------------------------------- /image_classification/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yangr116/ScalableViT/HEAD/image_classification/requirements.txt -------------------------------------------------------------------------------- /image_classification/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yangr116/ScalableViT/HEAD/image_classification/samplers.py -------------------------------------------------------------------------------- /image_classification/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yangr116/ScalableViT/HEAD/image_classification/utils.py -------------------------------------------------------------------------------- /object_detection/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yangr116/ScalableViT/HEAD/object_detection/README.md -------------------------------------------------------------------------------- /semantic_segmentation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yangr116/ScalableViT/HEAD/semantic_segmentation/README.md --------------------------------------------------------------------------------