├── .gitignore ├── README.md ├── distributed_train.sh ├── images ├── nlp_vis.png ├── overview.jpg └── vis.png ├── main.py ├── scripts ├── train_SoT_Base.sh ├── train_SoT_Small.sh └── train_SoT_Tiny.sh ├── setup.py └── sot_src ├── __init__.py ├── model ├── __init__.py ├── classifier.py ├── visualTokens.py └── vitBlock.py ├── normalization ├── MPN.py ├── __init__.py ├── ePN.py ├── scaleNorm.py └── svPN.py ├── representation ├── __init__.py ├── covariance.py └── multi_head_pooling.py └── sot.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiangtaoxie/SoT/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiangtaoxie/SoT/HEAD/README.md -------------------------------------------------------------------------------- /distributed_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiangtaoxie/SoT/HEAD/distributed_train.sh -------------------------------------------------------------------------------- /images/nlp_vis.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiangtaoxie/SoT/HEAD/images/nlp_vis.png -------------------------------------------------------------------------------- /images/overview.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiangtaoxie/SoT/HEAD/images/overview.jpg -------------------------------------------------------------------------------- /images/vis.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiangtaoxie/SoT/HEAD/images/vis.png -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiangtaoxie/SoT/HEAD/main.py -------------------------------------------------------------------------------- /scripts/train_SoT_Base.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiangtaoxie/SoT/HEAD/scripts/train_SoT_Base.sh -------------------------------------------------------------------------------- /scripts/train_SoT_Small.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiangtaoxie/SoT/HEAD/scripts/train_SoT_Small.sh -------------------------------------------------------------------------------- /scripts/train_SoT_Tiny.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiangtaoxie/SoT/HEAD/scripts/train_SoT_Tiny.sh -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiangtaoxie/SoT/HEAD/setup.py -------------------------------------------------------------------------------- /sot_src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiangtaoxie/SoT/HEAD/sot_src/__init__.py -------------------------------------------------------------------------------- /sot_src/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiangtaoxie/SoT/HEAD/sot_src/model/__init__.py -------------------------------------------------------------------------------- /sot_src/model/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiangtaoxie/SoT/HEAD/sot_src/model/classifier.py -------------------------------------------------------------------------------- /sot_src/model/visualTokens.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiangtaoxie/SoT/HEAD/sot_src/model/visualTokens.py -------------------------------------------------------------------------------- /sot_src/model/vitBlock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiangtaoxie/SoT/HEAD/sot_src/model/vitBlock.py -------------------------------------------------------------------------------- /sot_src/normalization/MPN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiangtaoxie/SoT/HEAD/sot_src/normalization/MPN.py -------------------------------------------------------------------------------- /sot_src/normalization/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiangtaoxie/SoT/HEAD/sot_src/normalization/__init__.py -------------------------------------------------------------------------------- /sot_src/normalization/ePN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiangtaoxie/SoT/HEAD/sot_src/normalization/ePN.py -------------------------------------------------------------------------------- /sot_src/normalization/scaleNorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiangtaoxie/SoT/HEAD/sot_src/normalization/scaleNorm.py -------------------------------------------------------------------------------- /sot_src/normalization/svPN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiangtaoxie/SoT/HEAD/sot_src/normalization/svPN.py -------------------------------------------------------------------------------- /sot_src/representation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiangtaoxie/SoT/HEAD/sot_src/representation/__init__.py -------------------------------------------------------------------------------- /sot_src/representation/covariance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiangtaoxie/SoT/HEAD/sot_src/representation/covariance.py -------------------------------------------------------------------------------- /sot_src/representation/multi_head_pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiangtaoxie/SoT/HEAD/sot_src/representation/multi_head_pooling.py -------------------------------------------------------------------------------- /sot_src/sot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiangtaoxie/SoT/HEAD/sot_src/sot.py --------------------------------------------------------------------------------