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 架构。