Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/MontaEllis/Pytorch-Medical-Segmentation
/ types & classes
Types & classes
69 in github.com/MontaEllis/Pytorch-Medical-Segmentation
⨍
Functions
168
◇
Types & classes
69
↓ 11 callers
Class
Conv3DBlock
models/three_d/unetr.py:27
↓ 10 callers
Class
DecoderBlock
models/two_d/unetpp.py:15
↓ 8 callers
Class
DownsamplerBlock
models/two_d/miniseg.py:75
↓ 7 callers
Class
ConvolutionalBlock
utils/convolution.py:12
↓ 6 callers
Class
Deconv3DBlock
models/three_d/unetr.py:40
↓ 6 callers
Class
DilatedParallelConvBlock
models/two_d/miniseg.py:36
↓ 5 callers
Class
SingleDeconv3DBlock
models/three_d/unetr.py:8
↓ 4 callers
Class
Down
models/two_d/unet.py:38
↓ 4 callers
Class
DownTransition
models/three_d/vnet3d.py:61
↓ 4 callers
Class
Up
models/two_d/unet.py:51
↓ 4 callers
Class
UpTransition
models/three_d/vnet3d.py:83
↓ 3 callers
Class
DilatedParallelConvBlockD2
models/two_d/miniseg.py:19
↓ 3 callers
Class
DoubleConv
(conv => BN => ReLU) * 2
models/two_d/unet.py:10
↓ 3 callers
Class
PSPUpsample
models/two_d/pspnet.py:159
↓ 3 callers
Class
SingleConv3DBlock
models/three_d/unetr.py:17
↓ 2 callers
Class
ConvBlock
models/two_d/miniseg.py:7
↓ 2 callers
Class
DilationBlock
utils/dilation.py:5
↓ 2 callers
Class
HighRes3DNet
models/three_d/highresnet.py:140
↓ 2 callers
Class
MiniSeg
models/two_d/miniseg.py:97
↓ 2 callers
Class
UNet
Implementations based on the Unet3D paper: https://arxiv.org/pdf/1706.00120.pdf
models/three_d/residual_unet3d.py:6
↓ 2 callers
Class
_DenseBlock
to keep the spatial dims o=i, this formula is applied o = [i + 2*p - k - (k-1)*(d-1)]/s + 1
models/three_d/densevoxelnet3d.py:36
↓ 1 callers
Class
ASPP
models/two_d/deeplab.py:127
↓ 1 callers
Class
DiceLoss
loss_function.py:101
↓ 1 callers
Class
Embeddings
models/three_d/unetr.py:128
↓ 1 callers
Class
InConv
models/two_d/unet.py:28
↓ 1 callers
Class
InputTransition
models/three_d/vnet3d.py:41
↓ 1 callers
Class
LUConv
models/three_d/vnet3d.py:21
↓ 1 callers
Class
MedData_test
data_function.py:160
↓ 1 callers
Class
MedData_train
data_function.py:39
↓ 1 callers
Class
OutConv
models/two_d/unet.py:75
↓ 1 callers
Class
OutputTransition
models/three_d/vnet3d.py:107
↓ 1 callers
Class
PSPModule
models/two_d/pspnet.py:139
↓ 1 callers
Class
Pad3d
utils/convolution.py:78
↓ 1 callers
Class
PositionwiseFeedForward
models/three_d/unetr.py:116
↓ 1 callers
Class
ResNet
models/two_d/deeplab.py:61
↓ 1 callers
Class
ResNet
models/two_d/pspnet.py:86
↓ 1 callers
Class
ResidualBlock
utils/residual.py:11
↓ 1 callers
Class
SelfAttention
models/three_d/unetr.py:54
↓ 1 callers
Class
Transformer
models/three_d/unetr.py:171
↓ 1 callers
Class
TransformerBlock
models/three_d/unetr.py:148
↓ 1 callers
Class
UNETR
models/three_d/unetr.py:194
↓ 1 callers
Class
_DenseBlock
models/three_d/densenet3d.py:29
↓ 1 callers
Class
_DenseLayer
models/three_d/densenet3d.py:7
↓ 1 callers
Class
_DenseLayer
models/three_d/densevoxelnet3d.py:17
↓ 1 callers
Class
_Transition
models/three_d/densenet3d.py:37
↓ 1 callers
Class
_Transition
models/three_d/densevoxelnet3d.py:49
↓ 1 callers
Class
_Upsampling
For transpose conv o = output, p = padding, k = kernel_size, s = stride, d = dilation o = (i -1)*s - 2*p + k + output_padding = (i-1)*2 +
models/three_d/densevoxelnet3d.py:65
Class
BasicBlock
models/two_d/pspnet.py:15
Class
BasicConv2d
models/two_d/unetpp.py:64
Class
BinaryDiceLoss
Dice loss of binary class Args: smooth: A float number to smooth loss, and avoid NaN error, default: 1 p: Denominator value: \sum{
loss_function.py:62
Class
Binary_Loss
loss_function.py:17
Class
Bottleneck
models/two_d/deeplab.py:15
Class
Bottleneck
models/two_d/pspnet.py:47
Class
DeepLabV3
models/two_d/deeplab.py:200
Class
DenseVoxelNet
Implementation based on https://arxiv.org/abs/1708.00573 Trainable params: 1,783,408 (roughly 1.8 mentioned in the paper)
models/three_d/densevoxelnet3d.py:90
Class
FCN32s
models/two_d/fcn.py:33
Class
FCN_Net
models/three_d/fcn3d.py:8
Class
HighRes2DNet
models/two_d/highresnet.py:140
Class
HighResNet
models/two_d/highresnet.py:13
Class
HighResNet
models/three_d/highresnet.py:13
Class
Mlp
models/three_d/unetr.py:102
Class
PSPNet
models/two_d/pspnet.py:174
Class
ResNet34UnetPlus
models/two_d/unetpp.py:84
Class
SegNet
models/two_d/segnet.py:7
Class
SkipDenseNet3D
Densely Connected Convolutional Networks" <https://arxiv.org/pdf/1608.06993.pdf>` Based on the implementation of https://github.com/tbuikr/3D-Skip
models/three_d/densenet3d.py:51
Class
UNet3D
models/three_d/unet3d.py:8
Class
Unet
models/two_d/unet.py:85
Class
VNet
Implementations based on the Vnet paper: https://arxiv.org/abs/1606.04797
models/three_d/vnet3d.py:124
Class
hparams
hparam.py:1