towhee.models.mpvit.mpvitΒΆ
Functions
Create MViT model. |
|
Generate drop path rate list following linear decay rule. |
|
mpvit_base : |
|
mpvit_small : |
|
mpvit_tiny : |
|
mpvit_xsmall : |
Classes
a linear layer for classification. |
|
Convolution with BN module. |
|
Convolutional Position Encoding. |
|
Convolutional relative position encoding. |
|
Depthwise Convolutional Patch Embedding layer Image to Patch Embedding. |
|
Depthwise Separable Convolution with BN module. |
|
Factorized attention with convolutional relative position encoding class. |
|
Multi-Head Convolutional self-Attention block. |
|
Multi-Head Convolutional self-Attention Encoder comprised of MHCA blocks. |
|
Multi-Head Convolutional self-Attention stage comprised of MHCAEncoder layers. |
|
Multi-Path ViT class. |
|
Feed-forward network (FFN, a.k.a.MLP) class. |
|
Depthwise Convolutional Patch Embedding stage comprised of DWCPatchEmbed layers. |
|
Residual block for convolutional local feature. |