快捷方式

SwinTransformer

SwinTransformer 模型基於 Swin Transformer:基於移位視窗的分層 Vision Transformer 論文。SwinTransformer V2 模型基於 Swin Transformer V2:擴大容量和解析度 論文。

模型構建器

以下模型構建器可用於例項化帶有或不帶有預訓練權重的 SwinTransformer 模型(原始版和 V2 版)。所有模型構建器內部都依賴於 torchvision.models.swin_transformer.SwinTransformer 基類。請參考 原始碼,瞭解有關此類的更多詳細資訊。

swin_t(*[, weights, progress])

根據 Swin Transformer:基於移位視窗的分層 Vision Transformer 構建 swin_tiny 架構。

swin_s(*[, weights, progress])

根據 Swin Transformer:基於移位視窗的分層 Vision Transformer 構建 swin_small 架構。

swin_b(*[, weights, progress])

根據 Swin Transformer:基於移位視窗的分層 Vision Transformer 構建 swin_base 架構。

swin_v2_t(*[, weights, progress])

根據 Swin Transformer V2:擴大容量和解析度 構建 swin_v2_tiny 架構。

swin_v2_s(*[, weights, progress])

根據 Swin Transformer V2:擴大容量和解析度 構建 swin_v2_small 架構。

swin_v2_b(*[, weights, progress])

根據 Swin Transformer V2:擴大容量和解析度 構建 swin_v2_base 架構。

文件

查閱 PyTorch 的全面開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深入教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源