MCPcopy Create free account

hub / github.com/HuCaoFighting/Swin-Unet / types & classes

Types & classes15 in github.com/HuCaoFighting/Swin-Unet

↓ 3 callersClassSynapse_dataset
datasets/dataset_synapse.py:50
↓ 2 callersClassPatchExpand
networks/swin_transformer_unet_skip_expand_decoder_sys.py:352
↓ 2 callersClassRandomGenerator
datasets/dataset_synapse.py:29
↓ 2 callersClassSwinTransformerBlock
r""" Swin Transformer Block. Args: dim (int): Number of input channels. input_resolution (tuple[int]): Input resulotion.
networks/swin_transformer_unet_skip_expand_decoder_sys.py:175
↓ 1 callersClassBasicLayer
A basic Swin Transformer layer for one stage. Args: dim (int): Number of input channels. input_resolution (tuple[int]): Input re
networks/swin_transformer_unet_skip_expand_decoder_sys.py:405
↓ 1 callersClassBasicLayer_up
A basic Swin Transformer layer for one stage. Args: dim (int): Number of input channels. input_resolution (tuple[int]): Input re
networks/swin_transformer_unet_skip_expand_decoder_sys.py:475
↓ 1 callersClassDiceLoss
utils.py:9
↓ 1 callersClassFinalPatchExpand_X4
networks/swin_transformer_unet_skip_expand_decoder_sys.py:377
↓ 1 callersClassMlp
networks/swin_transformer_unet_skip_expand_decoder_sys.py:26
↓ 1 callersClassPatchEmbed
r""" Image to Patch Embedding Args: img_size (int): Image size. Default: 224. patch_size (int): Patch token size. Default: 4.
networks/swin_transformer_unet_skip_expand_decoder_sys.py:534
↓ 1 callersClassSwinTransformerSys
r""" Swin Transformer A PyTorch impl of : `Swin Transformer: Hierarchical Vision Transformer using Shifted Windows` - https://arxiv
networks/swin_transformer_unet_skip_expand_decoder_sys.py:582
↓ 1 callersClassWindowAttention
r""" Window based multi-head self attention (W-MSA) module with relative position bias. It supports both of shifted and non-shifted window. A
networks/swin_transformer_unet_skip_expand_decoder_sys.py:77
ClassMoEFFNGating
networks/swin_transformer_unet_skip_expand_decoder_sys.py:8
ClassPatchMerging
r""" Patch Merging Layer. Args: input_resolution (tuple[int]): Resolution of input feature. dim (int): Number of input channels.
networks/swin_transformer_unet_skip_expand_decoder_sys.py:303
ClassSwinUnet
networks/vision_transformer.py:23