Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/UBC-CIC/COVID19-L3-Net
/ functions
Functions
166 in github.com/UBC-CIC/COVID19-L3-Net
⨍
Functions
166
◇
Types & classes
54
Method
__iter__
(self)
src/datasets/transformers/micnn_augmentor.py:164
Method
__len__
(self)
src/datasets/open_source.py:99
Method
__next__
(self)
src/datasets/transformers/micnn_augmentor.py:167
Function
apply_transform
(split, image, label=None, transform_name='basic', exp_dict=None)
src/datasets/transformers/__init__.py:13
Function
collate_fn
(batch)
src/utils.py:3
Function
compute_loss
(loss_name, logits, labels)
src/models/losses/__init__.py:6
Function
confusion_binary_class
(prediction, truth)
src/models/metrics.py:108
Function
dice_loss
Dice loss. :param input: The input (predicted) :param target: The target (ground truth) :returns: the Dice score between 0 and 1.
src/models/losses/dice_loss.py:5
Method
forward
(self, x)
src/models/base_networks/unet2d.py:21
Method
forward
(self, x)
src/models/base_networks/unet2d.py:35
Method
forward
(self, x)
src/models/base_networks/unet2d.py:73
Method
forward
(self, x)
src/models/base_networks/unet2d.py:96
Method
forward
(self, x)
src/models/base_networks/segmentation_models_pytorch/encoders/xception.py:40
Method
forward
(self, x)
src/models/base_networks/segmentation_models_pytorch/encoders/inceptionv4.py:66
Method
forward
(self, x)
src/models/base_networks/segmentation_models_pytorch/encoders/inceptionresnetv2.py:67
Method
forward
(self, x)
src/models/base_networks/segmentation_models_pytorch/encoders/timm_efficientnet.py:75
Method
forward
(self, x)
src/models/base_networks/segmentation_models_pytorch/encoders/senet.py:59
Method
forward
(self, x)
src/models/base_networks/segmentation_models_pytorch/encoders/dpn.py:56
Method
forward
(self, x)
src/models/base_networks/segmentation_models_pytorch/encoders/densenet.py:41
Method
forward
(self, x)
src/models/base_networks/segmentation_models_pytorch/encoders/densenet.py:72
Method
forward
(self, x)
src/models/base_networks/segmentation_models_pytorch/encoders/mobilenet.py:51
Method
forward
(self, x)
src/models/base_networks/segmentation_models_pytorch/encoders/vgg.py:66
Method
forward
(self, x)
src/models/base_networks/segmentation_models_pytorch/encoders/efficientnet.py:55
Method
forward
(self, x)
src/models/base_networks/segmentation_models_pytorch/encoders/resnet.py:56
Method
forward
(self, x)
src/models/base_networks/segmentation_models_pytorch/pspnet/decoder.py:19
Method
forward
(self, x)
src/models/base_networks/segmentation_models_pytorch/pspnet/decoder.py:34
Method
forward
(self, *features)
src/models/base_networks/segmentation_models_pytorch/pspnet/decoder.py:66
Method
forward
(self, x)
src/models/base_networks/segmentation_models_pytorch/base/modules.py:62
Method
forward
(self, x)
src/models/base_networks/segmentation_models_pytorch/base/modules.py:72
Method
forward
(self, x)
src/models/base_networks/segmentation_models_pytorch/base/modules.py:101
Method
forward
(self, x)
src/models/base_networks/segmentation_models_pytorch/base/modules.py:117
Method
forward
(self, x)
src/models/base_networks/segmentation_models_pytorch/base/modules.py:122
Method
get_avg_score
(self)
src/models/metrics.py:69
Function
get_base
(base_name, exp_dict, n_classes)
src/models/base_networks/__init__.py:5
Function
get_class_map
(n_classes)
src/datasets/transformers/trans_utils.py:8
Function
get_dataset
(dataset_dict, split, datadir, exp_dict, dataset_size=None)
src/datasets/__init__.py:20
Function
get_encoder_names
()
src/models/base_networks/segmentation_models_pytorch/encoders/__init__.py:47
Function
get_model
(model_dict, exp_dict=None, train_set=None)
src/models/__init__.py:9
Function
get_normalization_stats
(windows)
src/datasets/transformers/trans_utils.py:63
Function
get_preprocessing_fn
(encoder_name, pretrained="imagenet")
src/models/base_networks/segmentation_models_pytorch/encoders/__init__.py:65
Function
get_thresholds_stats
(windows)
src/datasets/transformers/trans_utils.py:42
Function
initialize_decoder
(module)
src/models/base_networks/segmentation_models_pytorch/base/initialization.py:4
Function
initialize_head
(module)
src/models/base_networks/segmentation_models_pytorch/base/initialization.py:22
Method
load_state_dict
(self, state_dict)
src/models/base_networks/segmentation_models_pytorch/encoders/xception.py:50
Method
load_state_dict
(self, state_dict, **kwargs)
src/models/base_networks/segmentation_models_pytorch/encoders/inceptionv4.py:77
Method
load_state_dict
(self, state_dict, **kwargs)
src/models/base_networks/segmentation_models_pytorch/encoders/inceptionresnetv2.py:78
Method
load_state_dict
(self, state_dict, **kwargs)
src/models/base_networks/segmentation_models_pytorch/encoders/timm_efficientnet.py:85
Method
load_state_dict
(self, state_dict, **kwargs)
src/models/base_networks/segmentation_models_pytorch/encoders/senet.py:69
Method
load_state_dict
(self, state_dict, **kwargs)
src/models/base_networks/segmentation_models_pytorch/encoders/dpn.py:70
Method
load_state_dict
(self, state_dict)
src/models/base_networks/segmentation_models_pytorch/encoders/densenet.py:87
Method
load_state_dict
(self, state_dict, **kwargs)
src/models/base_networks/segmentation_models_pytorch/encoders/mobilenet.py:61
Method
load_state_dict
(self, state_dict, **kwargs)
src/models/base_networks/segmentation_models_pytorch/encoders/efficientnet.py:79
Method
load_state_dict
(self, state_dict, **kwargs)
src/models/base_networks/segmentation_models_pytorch/encoders/resnet.py:66
Method
make_dilated
(self, stage_list, dilation_list)
src/models/base_networks/segmentation_models_pytorch/encoders/xception.py:25
Method
make_dilated
(self, stage_list, dilation_list)
src/models/base_networks/segmentation_models_pytorch/encoders/inceptionv4.py:52
Method
make_dilated
(self, stage_list, dilation_list)
src/models/base_networks/segmentation_models_pytorch/encoders/_base.py:35
Method
make_dilated
(self, stage_list, dilation_list)
src/models/base_networks/segmentation_models_pytorch/encoders/inceptionresnetv2.py:53
Method
make_dilated
(self, stage_list, dilation_list)
src/models/base_networks/segmentation_models_pytorch/encoders/densenet.py:57
Method
make_dilated
(self, stage_list, dilation_list)
src/models/base_networks/segmentation_models_pytorch/encoders/vgg.py:51
Method
out_channels
Return channels dimensions for each tensor of forward output of encoder
src/models/base_networks/segmentation_models_pytorch/encoders/_base.py:16
Function
patch_first_conv
Change first convolution layer input channels. In case: in_channels == 1 or in_channels == 2 -> reuse original weights in_channels
src/models/base_networks/segmentation_models_pytorch/encoders/_utils.py:5
Method
predict
Inference method. Switch model to `eval` mode, call `.forward(x)` with `torch.no_grad()` Args: x: 4D torch tensor with shape (bat
src/models/base_networks/segmentation_models_pytorch/base/model.py:26
Function
preprocess_input
( x, mean=None, std=None, input_space="RGB", input_range=None, **kwargs )
src/models/base_networks/segmentation_models_pytorch/encoders/_preprocessing.py:4
Function
replace_strides_with_dilation
Patch Conv2d modules replacing strides with dilation
src/models/base_networks/segmentation_models_pytorch/encoders/_utils.py:39
Method
set_thread_id
(self, thread_id)
src/datasets/transformers/micnn_augmentor.py:172
Method
val_on_batch
(self, model, batch)
src/models/metrics.py:52
← previous
101–166 of 166, ranked by callers