Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/black0017/MedicalZooPytorch
/ types & classes
Types & classes
94 in github.com/black0017/MedicalZooPytorch
⨍
Functions
419
◇
Types & classes
94
↓ 15 callers
Class
GreenBlock
lib/medzoo/ResNet3D_VAE.py:11
↓ 13 callers
Class
DiceLoss
Computes Dice Loss according to https://arxiv.org/abs/1606.04797. For multi-class segmentation `weight` parameter can be used to assign different
lib/losses3D/dice.py:6
↓ 8 callers
Class
MRIDatasetMRBRAINS2018
lib/medloaders/mrbrains2018.py:13
↓ 8 callers
Class
SinglePathDenseNet
lib/medzoo/Densenet3D.py:74
↓ 7 callers
Class
DownTransition
lib/medzoo/Vnet.py:65
↓ 7 callers
Class
ResNetMed3D
lib/medzoo/ResNet3DMedNet.py:175
↓ 7 callers
Class
UpBlock2
lib/medzoo/ResNet3D_VAE.py:78
↓ 7 callers
Class
UpTransition
lib/medzoo/Vnet.py:87
↓ 7 callers
Class
_HyperDenseLayer
lib/medzoo/Densenet3D.py:12
↓ 6 callers
Class
MICCAIBraTS2018
Code for reading the infant brain MICCAIBraTS2018 challenge
lib/medloaders/brats2018.py:14
↓ 5 callers
Class
Conv1x1x1
lib/medzoo/HighResNet3D.py:76
↓ 4 callers
Class
COVID_Seg_Dataset
Code for reading the COVID Segmentation dataset Segmentation Task 1: Learning with limited annotations This task is based on the CO
lib/medloaders/Covid_Segmentation_dataset.py:12
↓ 4 callers
Class
COVIDxDataset
Code for reading the COVIDxDataset
lib/medloaders/COVIDxdataset.py:13
↓ 4 callers
Class
CovidCTDataset
lib/medloaders/covid_ct_dataset.py:9
↓ 4 callers
Class
Down
lib/medzoo/Unet2D.py:36
↓ 4 callers
Class
MICCAI2019_gleason_pathology
Code for reading Gleason 2019 MICCAI Challenge
lib/medloaders/miccai_2019_pathology.py:14
↓ 4 callers
Class
MRIDatasetISEG2019
Code for reading the infant brain MRI dataset of ISEG 2017 challenge
lib/medloaders/iseg2019.py:14
↓ 4 callers
Class
Up
lib/medzoo/Unet2D.py:49
↓ 3 callers
Class
BlueBlock
lib/medzoo/ResNet3D_VAE.py:53
↓ 3 callers
Class
ConvRed
lib/medzoo/HighResNet3D.py:30
↓ 3 callers
Class
DilatedConv2
lib/medzoo/HighResNet3D.py:45
↓ 3 callers
Class
DilatedConv4
lib/medzoo/HighResNet3D.py:60
↓ 3 callers
Class
DoubleConv
(conv => BN => ReLU) * 2
lib/medzoo/Unet2D.py:8
↓ 3 callers
Class
DownBlock
lib/medzoo/ResNet3D_VAE.py:42
↓ 3 callers
Class
MRIDatasetISEG2017
Code for reading the infant brain MRI dataset of ISEG 2017 challenge
lib/medloaders/iseg2017.py:14
↓ 3 callers
Class
TensorboardWriter
lib/visual3D_temp/BaseWriter.py:20
↓ 2 callers
Class
BCEDiceLoss
Linear combination of BCE and Dice losses3D
lib/losses3D/BCE_dice.py:7
↓ 2 callers
Class
ConvInit
lib/medzoo/HighResNet3D.py:11
↓ 2 callers
Class
CovidNet
lib/medzoo/COVIDNet.py:48
↓ 2 callers
Class
Decoder
lib/medzoo/ResNet3D_VAE.py:151
↓ 2 callers
Class
GeneralizedDiceLoss
Computes Generalized Dice Loss (GDL) as described in https://arxiv.org/pdf/1707.03237.pdf.
lib/losses3D/generalized_dice.py:8
↓ 2 callers
Class
IXIMRIdataset
Code for reading the IXI brain MRI dataset This loader is implemented for cross-dataset testing
lib/medloaders/ixi_t1_t2.py:12
↓ 2 callers
Class
InputTransition
lib/medzoo/Vnet.py:45
↓ 2 callers
Class
MICCAIBraTS2019
Code for reading the infant brain MICCAIBraTS2018 challenge
lib/medloaders/brats2019.py:14
↓ 2 callers
Class
MICCAIBraTS2020
Code for reading the infant brain MICCAIBraTS2018 challenge
lib/medloaders/brats2020.py:14
↓ 2 callers
Class
MetricTracker
lib/utils/covid_utils.py:34
↓ 2 callers
Class
OutputTransition
lib/medzoo/Vnet.py:111
↓ 2 callers
Class
PixelWiseCrossEntropyLoss
lib/losses3D/pixel_wise_cross_entropy.py:6
↓ 2 callers
Class
ResNetEncoder
lib/medzoo/ResNet3D_VAE.py:97
↓ 2 callers
Class
TagsAngularLoss
lib/losses3D/tags_angular_loss.py:8
↓ 2 callers
Class
VAE
lib/medzoo/ResNet3D_VAE.py:183
↓ 2 callers
Class
WeightedCrossEntropyLoss
WeightedCrossEntropyLoss (WCE) as described in https://arxiv.org/pdf/1707.03237.pdf
lib/losses3D/weight_cross_entropy.py:5
↓ 2 callers
Class
WeightedSmoothL1Loss
lib/losses3D/weight_smooth_l1.py:7
↓ 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
lib/medzoo/DenseVoxelNet.py:45
↓ 1 callers
Class
CNN
lib/medzoo/COVIDNet.py:176
↓ 1 callers
Class
ContrastiveLoss
Implementation of contrastive loss defined in https://arxiv.org/pdf/1708.02551.pdf 'Semantic Instance Segmentation with a Discriminative Loss
lib/losses3D/ContrastiveLoss.py:7
↓ 1 callers
Class
DenseVoxelNet
Implementation based on https://arxiv.org/abs/1708.00573 Trainable params: 1,783,408 (roughly 1.8 mentioned in the paper)
lib/medzoo/DenseVoxelNet.py:99
↓ 1 callers
Class
DiceLoss2D
lib/losses3D/Dice2D.py:6
↓ 1 callers
Class
DualPathDenseNet
lib/medzoo/Densenet3D.py:145
↓ 1 callers
Class
DualSingleDenseNet
2-stream and 3-stream implementation with early fusion dual-single-densenet OR Disentangled modalities with early fusion in the paper
lib/medzoo/Densenet3D.py:232
↓ 1 callers
Class
Flatten
lib/medzoo/COVIDNet.py:7
↓ 1 callers
Class
HighResNet3D
lib/medzoo/HighResNet3D.py:92
↓ 1 callers
Class
HyperDenseNet
lib/medzoo/HyperDensenet.py:424
↓ 1 callers
Class
HyperDenseNet_2Mod
lib/medzoo/HyperDensenet.py:273
↓ 1 callers
Class
InConv
lib/medzoo/Unet2D.py:26
↓ 1 callers
Class
LUConv
lib/medzoo/Vnet.py:25
↓ 1 callers
Class
OutConv
lib/medzoo/Unet2D.py:73
↓ 1 callers
Class
ResNet3dVAE
lib/medzoo/ResNet3D_VAE.py:253
↓ 1 callers
Class
ResidualConv
lib/medzoo/HyperDensenet.py:30
↓ 1 callers
Class
SkipDenseNet3D
Densely Connected Convolutional Networks" <https://arxiv.org/pdf/1608.06993.pdf>` Based on the implementation of https://github.com/tbuikr/3D-Skip
lib/medzoo/SkipDenseNet3D.py:59
↓ 1 callers
Class
TestCriterion
tests/test_losses3D.py:17
↓ 1 callers
Class
TestDataLoaders
tests/test_dataloaders.py:9
↓ 1 callers
Class
Trainer
Trainer class
lib/train/trainer.py:8
↓ 1 callers
Class
TranspConvNet
(segmentation)we transfer encoder part from Med3D as the feature extraction part and then segmented lung in whole body followed by three gro
lib/medzoo/ResNet3DMedNet.py:129
↓ 1 callers
Class
UNet3D
Implementations based on the Unet3D paper: https://arxiv.org/abs/1606.06650
lib/medzoo/Unet3D.py:8
↓ 1 callers
Class
Unet
lib/medzoo/Unet2D.py:83
↓ 1 callers
Class
VNet
Implementations based on the Vnet paper: https://arxiv.org/abs/1606.04797
lib/medzoo/Vnet.py:128
↓ 1 callers
Class
VNetLight
A lighter version of Vnet that skips down_tr256 and up_tr256 in oreder to reduce time and space complexity
lib/medzoo/Vnet.py:173
↓ 1 callers
Class
_DenseBlock
lib/medzoo/SkipDenseNet3D.py:37
↓ 1 callers
Class
_DenseLayer
lib/medzoo/SkipDenseNet3D.py:15
↓ 1 callers
Class
_DenseLayer
lib/medzoo/DenseVoxelNet.py:26
↓ 1 callers
Class
_HyperDenseBlock
Constructs a series of dense-layers based on in and out kernels list
lib/medzoo/Densenet3D.py:32
↓ 1 callers
Class
_HyperDenseBlockEarlyFusion
lib/medzoo/Densenet3D.py:55
↓ 1 callers
Class
_Transition
lib/medzoo/SkipDenseNet3D.py:45
↓ 1 callers
Class
_Transition
lib/medzoo/DenseVoxelNet.py:58
↓ 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 +
lib/medzoo/DenseVoxelNet.py:74
Class
BaseModel
r""" BaseModel with basic functionalities for checkpointing and restoration.
lib/medzoo/BaseModelClass.py:12
Class
BaseTrainer
Base class for all trainers
lib/train/BaseTrainer.py:7
Class
BasicBlock
lib/medzoo/ResNet3DMedNet.py:58
Class
Bottleneck
lib/medzoo/ResNet3DMedNet.py:90
Class
ComposeTransforms
Composes several transforms together.
lib/augment3D/__init__.py:44
Class
ElasticTransform
lib/augment3D/elastic_deform.py:77
Class
GaussianNoise
lib/augment3D/gaussian_noise.py:10
Class
PEPX
lib/medzoo/COVIDNet.py:12
Class
RandomChoice
choose a random tranform from list an apply transforms: tranforms to apply p: probability
lib/augment3D/__init__.py:16
Class
RandomCropToLabels
lib/augment3D/random_crop.py:32
Class
RandomFlip
lib/augment3D/random_flip.py:27
Class
RandomRotation
lib/augment3D/random_rotate.py:23
Class
RandomShift
lib/augment3D/random_shift.py:19
Class
RandomZoom
lib/augment3D/random_rescale.py:20
Class
SkipLastTargetChannelWrapper
Loss wrapper which removes additional target channel
lib/losses3D/__init__.py:58
Class
UpBlock1
TODO fix transpose conv to double spatial dim
lib/medzoo/ResNet3D_VAE.py:64
Class
_AbstractDiceLoss
Base class for different implementations of Dice loss.
lib/losses3D/BaseClass.py:10
Class
_MaskingLossWrapper
Loss wrapper which prevents the gradient of the loss to be computed where target is equal to `ignore_index`.
lib/losses3D/__init__.py:80