def __init__(self, dim, window_size,接下来定义Swim-unet模型,包括Encoder和Decoder两部分。其中,Encoder部分采用Swin Transformer模块进行特征提取和上采样,并输出多尺度的特征图;Decoder部分则采用U-Net结构进行特征融合和下采样,并输出最终的分割结果。 ```python # 定义Swim-unet模型 class SwinUnet(nn.Module...
总结:实际上由于SwinUnet是一个encoder-decoder对称的结构,因此加载权重时,作者并没有像通常那样仅仅加载encoder部分而不加载decoder部分,而是同时将encoder的权重对称地加载到了decoder上(除了swin_unet.layers_up.1/2/3.upsample) 详细注释看下面: defload_from(self,args):pretrained_path=args.pretrain_ckpt#config...