CSWinTransformer

March 29, 2022 ยท View on GitHub


Catalogue

1. Overview

CSWinTransformer is a new visual Transformer network that can be used as a general backbone network in the field of computer vision. CSWinTransformer proposes to do self-attention through a cross-shaped window, which not only has a very high computational efficiency, but also can obtain a global receptive field through two-layer calculation. CSWinTransformer also proposed a new encoding method: LePE, which further improved the accuracy of the model. Paper

2. Accuracy, FLOPs and Parameters

ModelsTop1Top5Reference
top1
Reference
top5
FLOPs
(G)
Params
(M)
CSWinTransformer_tiny_2240.82810.96280.828-4.122
CSWinTransformer_small_2240.83580.96580.836-6.435
CSWinTransformer_base_2240.84200.96920.842-14.377
CSWinTransformer_large_2240.86430.97990.865-32.2173.3
CSWinTransformer_base_3840.85500.97490.855-42.277
CSWinTransformer_large_3840.87480.98330.875-94.7173.3