SwinTransformer

SwinTransformer 模型基於 Swin Transformer: Hierarchical Vision Transformer using Shifted Windows 論文。SwinTransformer V2 模型基於 Swin Transformer V2: Scaling Up Capacity and Resolution 論文。

模型構建器

以下模型構建器可用於例項化具有和不具有預訓練權重的 SwinTransformer 模型（原始版本和 V2 版本）。所有模型構建器內部都依賴於 torchvision.models.swin_transformer.SwinTransformer 基類。有關此類更多詳細資訊，請參閱 原始碼。

swin_t(*[, weights, progress])	從 Swin Transformer: Hierarchical Vision Transformer using Shifted Windows 構建 swin_tiny 架構。
swin_s(*[, weights, progress])	從 Swin Transformer: Hierarchical Vision Transformer using Shifted Windows 構建 swin_small 架構。
swin_b(*[, weights, progress])	從 Swin Transformer: Hierarchical Vision Transformer using Shifted Windows 構建 swin_base 架構。
swin_v2_t(*[, weights, progress])	從 Swin Transformer V2: Scaling Up Capacity and Resolution 構建 swin_v2_tiny 架構。
swin_v2_s(*[, weights, progress])	從 Swin Transformer V2: Scaling Up Capacity and Resolution 構建 swin_v2_small 架構。
swin_v2_b(*[, weights, progress])	從 Swin Transformer V2: Scaling Up Capacity and Resolution 構建 swin_v2_base 架構。