├── README.md ├── requirements.txt ├── train.py ├── vit.png ├── vit.py └── weights └── model.09-1.15.h5 /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tuvovan/Vision_Transformer_Keras/HEAD/README.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | tensorflow==2.2.0+ 2 | keras 2.3.0 3 | pillow 4 | numpy 5 | -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tuvovan/Vision_Transformer_Keras/HEAD/train.py -------------------------------------------------------------------------------- /vit.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tuvovan/Vision_Transformer_Keras/HEAD/vit.png -------------------------------------------------------------------------------- /vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tuvovan/Vision_Transformer_Keras/HEAD/vit.py -------------------------------------------------------------------------------- /weights/model.09-1.15.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tuvovan/Vision_Transformer_Keras/HEAD/weights/model.09-1.15.h5 --------------------------------------------------------------------------------