Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/HuCaoFighting/Swin-Unet
/ types & classes
Types & classes
15 in github.com/HuCaoFighting/Swin-Unet
⨍
Functions
65
◇
Types & classes
15
↓ 3 callers
Class
Synapse_dataset
datasets/dataset_synapse.py:50
↓ 2 callers
Class
PatchExpand
networks/swin_transformer_unet_skip_expand_decoder_sys.py:352
↓ 2 callers
Class
RandomGenerator
datasets/dataset_synapse.py:29
↓ 2 callers
Class
SwinTransformerBlock
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 callers
Class
BasicLayer
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 callers
Class
BasicLayer_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 callers
Class
DiceLoss
utils.py:9
↓ 1 callers
Class
FinalPatchExpand_X4
networks/swin_transformer_unet_skip_expand_decoder_sys.py:377
↓ 1 callers
Class
Mlp
networks/swin_transformer_unet_skip_expand_decoder_sys.py:26
↓ 1 callers
Class
PatchEmbed
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 callers
Class
SwinTransformerSys
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 callers
Class
WindowAttention
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
Class
MoEFFNGating
networks/swin_transformer_unet_skip_expand_decoder_sys.py:8
Class
PatchMerging
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
Class
SwinUnet
networks/vision_transformer.py:23