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 Transformer: Hierarchical Vision Transformer using Shifted Windows构建一个swin_tiny架构。 |
|
从Swin Transformer: Hierarchical Vision Transformer using Shifted Windows构建一个swin_small架构。 |
|
从Swin Transformer: Hierarchical Vision Transformer using Shifted Windows构建一个swin_base架构。 |
|
从Swin Transformer V2: Scaling Up Capacity and Resolution构建一个swin_v2_tiny架构。 |
|
从Swin Transformer V2: Scaling Up Capacity and Resolution构建一个swin_v2_small架构。 |
|
从Swin Transformer V2: Scaling Up Capacity and Resolution构建一个swin_v2_base架构。 |