Shortcuts

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(*[, 权重, 进度])

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(*[, 权重, 进度])

Swin Transformer V2: Scaling Up Capacity and Resolution构建一个swin_v2_tiny架构。

swin_v2_s(*[, 权重, 进度])

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