├── LICENSE ├── README.md ├── dist_test.sh ├── dist_train.sh ├── images ├── Swiftformer_performance.png ├── attention_comparison.pdf ├── attentions_comparison.png ├── detection_seg.png └── semantic_seg.png ├── main.py ├── models ├── __init__.py └── swiftformer.py ├── requirements.txt ├── slurm_train.sh └── util ├── __init__.py ├── datasets.py ├── engine.py ├── losses.py ├── samplers.py └── utils.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Amshaker/SwiftFormer/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Amshaker/SwiftFormer/HEAD/README.md -------------------------------------------------------------------------------- /dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Amshaker/SwiftFormer/HEAD/dist_test.sh -------------------------------------------------------------------------------- /dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Amshaker/SwiftFormer/HEAD/dist_train.sh -------------------------------------------------------------------------------- /images/Swiftformer_performance.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Amshaker/SwiftFormer/HEAD/images/Swiftformer_performance.png -------------------------------------------------------------------------------- /images/attention_comparison.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Amshaker/SwiftFormer/HEAD/images/attention_comparison.pdf -------------------------------------------------------------------------------- /images/attentions_comparison.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Amshaker/SwiftFormer/HEAD/images/attentions_comparison.png -------------------------------------------------------------------------------- /images/detection_seg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Amshaker/SwiftFormer/HEAD/images/detection_seg.png -------------------------------------------------------------------------------- /images/semantic_seg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Amshaker/SwiftFormer/HEAD/images/semantic_seg.png -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Amshaker/SwiftFormer/HEAD/main.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Amshaker/SwiftFormer/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/swiftformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Amshaker/SwiftFormer/HEAD/models/swiftformer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Amshaker/SwiftFormer/HEAD/requirements.txt -------------------------------------------------------------------------------- /slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Amshaker/SwiftFormer/HEAD/slurm_train.sh -------------------------------------------------------------------------------- /util/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Amshaker/SwiftFormer/HEAD/util/__init__.py -------------------------------------------------------------------------------- /util/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Amshaker/SwiftFormer/HEAD/util/datasets.py -------------------------------------------------------------------------------- /util/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Amshaker/SwiftFormer/HEAD/util/engine.py -------------------------------------------------------------------------------- /util/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Amshaker/SwiftFormer/HEAD/util/losses.py -------------------------------------------------------------------------------- /util/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Amshaker/SwiftFormer/HEAD/util/samplers.py -------------------------------------------------------------------------------- /util/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Amshaker/SwiftFormer/HEAD/util/utils.py --------------------------------------------------------------------------------