├── README.md ├── datasets ├── Pets.py ├── STL10.py ├── datasets_utils.py └── load_dataset.py ├── imgs └── SiT_.png ├── main.py ├── requirements.txt ├── utils.py └── vision_transformer.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sara-Ahmed/SiT/HEAD/README.md -------------------------------------------------------------------------------- /datasets/Pets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sara-Ahmed/SiT/HEAD/datasets/Pets.py -------------------------------------------------------------------------------- /datasets/STL10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sara-Ahmed/SiT/HEAD/datasets/STL10.py -------------------------------------------------------------------------------- /datasets/datasets_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sara-Ahmed/SiT/HEAD/datasets/datasets_utils.py -------------------------------------------------------------------------------- /datasets/load_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sara-Ahmed/SiT/HEAD/datasets/load_dataset.py -------------------------------------------------------------------------------- /imgs/SiT_.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sara-Ahmed/SiT/HEAD/imgs/SiT_.png -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sara-Ahmed/SiT/HEAD/main.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | timm==0.3.2 2 | kornia 3 | pandas 4 | imageio 5 | tabulate 6 | colorama 7 | 8 | 9 | -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sara-Ahmed/SiT/HEAD/utils.py -------------------------------------------------------------------------------- /vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Sara-Ahmed/SiT/HEAD/vision_transformer.py --------------------------------------------------------------------------------