├── README.md ├── class_indices.json ├── flops.py ├── my_dataset.py ├── predict.py ├── train.py ├── utils.py ├── vit_model.py └── weights └── README.md /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zht8506/Vision-Transformer-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /class_indices.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zht8506/Vision-Transformer-pytorch/HEAD/class_indices.json -------------------------------------------------------------------------------- /flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zht8506/Vision-Transformer-pytorch/HEAD/flops.py -------------------------------------------------------------------------------- /my_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zht8506/Vision-Transformer-pytorch/HEAD/my_dataset.py -------------------------------------------------------------------------------- /predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zht8506/Vision-Transformer-pytorch/HEAD/predict.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zht8506/Vision-Transformer-pytorch/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zht8506/Vision-Transformer-pytorch/HEAD/utils.py -------------------------------------------------------------------------------- /vit_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zht8506/Vision-Transformer-pytorch/HEAD/vit_model.py -------------------------------------------------------------------------------- /weights/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zht8506/Vision-Transformer-pytorch/HEAD/weights/README.md --------------------------------------------------------------------------------