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