Functions — github.com/HiLab-git/SSL4MIS

hub / github.com/HiLab-git/SSL4MIS / functions

Functions575 in github.com/HiLab-git/SSL4MIS

⨍ Functions 575◇ Types & classes 123

↓ 81 callersMethodstate_dict

(self)

code/augmentations/__init__.py:13

↓ 41 callersFunctiondice_loss

(score, target)

code/utils/losses.py:8

↓ 22 callersMethodget_device

(self)

code/networks/neural_network.py:54

↓ 20 callersFunctioninit_weights

(net, init_type='normal')

code/networks/networks_other.py:64

↓ 20 callersMethodlog

(self, train_point)

code/utils/util.py:199

↓ 18 callersMethod__init__

(self, n_features, reduction=6)

code/networks/utils.py:282

↓ 15 callersFunctionnet_factory

(net_type="unet", in_chns=1, class_num=3)

code/networks/net_factory.py:75

↓ 15 callersFunctiontest_single_volume

(image, label, net, classes, patch_size=[256, 256])

code/val_2D.py:18

↓ 11 callersMethod__init__

(self, in_chns, class_num)

code/networks/unet.py:305

↓ 10 callersMethod_compute_steps_for_sliding_window

(patch_size: Tuple[int, ...], image_size: Tuple[int, ...], step_size: float)

code/networks/neural_network.py:292

↓ 10 callersFunctionnet_factory_3d

(net_type="unet_3D", in_chns=1, class_num=2)

code/networks/net_factory_3d.py:8

↓ 10 callersFunctiontest_all_case

(net, base_dir, test_list="full_test.list", num_classes=4, patch_size=(48, 160, 160), stride_xy=32, stride_z=2

code/val_3D.py:91

↓ 9 callersMethod__init__

(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)

code/networks/swin_transformer_unet_skip_expand_decoder_sys.py:10

↓ 8 callersFunction_get_pretrained_settings

(encoder)

code/networks/encoder_tool.py:108

↓ 8 callersFunction_get_pretrained_settings

(encoder)

code/networks/efficient_encoder.py:111

↓ 6 callersMethodencoder

(self, input)

code/networks/vnet.py:180

↓ 6 callersMethodload_state_dict

(self, state)

code/augmentations/__init__.py:8

↓ 5 callersMethod__init__

(self, use_lsgan=True, target_real_label=1.0, target_fake_label=0.0, tensor=torch.FloatTensor

code/networks/networks_other.py:261

↓ 5 callersMethod__init__

(self, in_channels, gating_channels, inter_channels=None, mode='concatenation', sub_sample_fa

code/networks/grid_attention_layer.py:163

↓ 5 callersMethod__init__

(self, n_channels=3, n_classes=2, n_filters=16, normalization='none', has_dropout=False)

code/networks/vnet.py:146

↓ 5 callersMethoddecoder

(self, features)

code/networks/vnet.py:202

↓ 5 callersMethodupdate

(self, val, n=1)

code/utils/util.py:176

↓ 4 callersMethod__init__

(self, in_channels, num_classes, encoder_relu=False, decoder_relu=True)

code/networks/enet.py:465

↓ 4 callersMethod__init__

(self, name, **params)

code/networks/attention.py:94

↓ 4 callersMethod__init__

(self, in_channels)

code/networks/VoxResNet.py:27

↓ 4 callersFunction_enhance

(x, op, level)

code/augmentations/ctaugment.py:105

↓ 4 callersFunctionmaybe_to_torch

(d)

code/networks/neural_network.py:33

↓ 4 callersMethodpolicy

(self, probe, weak)

code/augmentations/ctaugment.py:54

↓ 4 callersFunctionto_cuda

(data, non_blocking=True, gpu_id=0)

code/networks/neural_network.py:42

↓ 3 callersMethod__init__

( self, encoder_name: str = "resnet34", encoder_depth: int = 5,

code/networks/efficientunet.py:179

↓ 3 callersMethod__init__

(self, size=None, scale_factor=None, mode='nearest', align_corners=False)

code/networks/nnunet.py:174

↓ 3 callersMethod__init__

(self, in_chns, out_chns, num_filters, ratios)

code/networks/pnet.py:88

↓ 3 callersFunction_imageop

(x, op, level)

code/augmentations/ctaugment.py:109

↓ 3 callersMethod_internal_predict_2D_2Dconv

This one does fully convolutional inference. No sliding window

code/networks/neural_network.py:477

↓ 3 callersFunctioncalculate_metric_percase

(pred, gt)

code/test_2D_fully.py:31

↓ 3 callersMethodcta_apply

(self, pil_img, ops)

code/dataloaders/dataset.py:135

↓ 3 callersMethodflops

(self)

code/networks/swin_transformer_unet_skip_expand_decoder_sys.py:477

↓ 3 callersMethodload_state_dict

(self, state_dict, **kwargs)

code/networks/encoder_tool.py:102

↓ 3 callersMethodrate_to_p

(self, rate)

code/augmentations/ctaugment.py:48

↓ 3 callersMethodresize

(self, image)

code/dataloaders/dataset.py:142

↓ 2 callersFunctionDropout

(x, p=0.3)

code/networks/unet.py:272

↓ 2 callersFunctionFeatureDropout

(x)

code/networks/unet.py:277

↓ 2 callersFunction_filter

(x, op, level)

code/augmentations/ctaugment.py:113

↓ 2 callersMethod_get_gaussian

(patch_size, sigma_scale=1. / 8)

code/networks/neural_network.py:274

↓ 2 callersMethod_internal_maybe_mirror_and_pred_2D

(self, x: Union[np.ndarray, torch.tensor], mirror_axes: tuple, do_m

code/networks/neural_network.py:613

↓ 2 callersMethod_internal_maybe_mirror_and_pred_3D

(self, x: Union[np.ndarray, torch.tensor], mirror_axes: tuple, do_m

code/networks/neural_network.py:553

↓ 2 callersMethod_internal_predict_2D_2Dconv_tiled

(self, x: np.ndarray, step_size: float, do_mirroring: bool, mirror_axes: tuple,

code/networks/neural_network.py:656

↓ 2 callersFunctionbrain_bbox

(data, gt)

code/dataloaders/brats_proprecessing.py:10

↓ 2 callersFunctioncalculate_metric_percase

(pred, gt)

code/val_2D.py:7

↓ 2 callersFunctioncreate_model

(ema=False)

code/train_cross_pseudo_supervision_2D.py:115

↓ 2 callersFunctioncreate_model

(ema=False)

code/train_uncertainty_aware_mean_teacher_3D.py:86

↓ 2 callersFunctioncreate_model

(ema=False)

code/train_fixmatch_cta.py:142

↓ 2 callersFunctioncreate_model

(ema=False)

code/train_mean_teacher_2D.py:98

↓ 2 callersFunctioncreate_model

(ema=False)

code/train_interpolation_consistency_training_2D.py:100

↓ 2 callersFunctioncreate_model

(ema=False)

code/train_mean_teacher_3D.py:86

↓ 2 callersFunctioncreate_model

(ema=False)

code/train_regularized_dropout_2D.py:115

↓ 2 callersFunctioncreate_model

(ema=False)

code/train_fixmatch_standard_augs.py:138

↓ 2 callersFunctioncreate_model

(ema=False)

code/train_interpolation_consistency_training_3D.py:90

↓ 2 callersFunctioncreate_model

(ema=False)

code/train_uncertainty_aware_mean_teacher_2D.py:98

↓ 2 callersFunctionget_norm_layer

(norm_type='instance')

code/networks/networks_other.py:78

↓ 2 callersFunctionget_pascal_labels

Load the mapping that associates pascal classes with label colors Returns: np.ndarray with dimensions (21, 3)

code/dataloaders/utils.py:42

↓ 2 callersFunctiongrouper

Collect data into fixed-length chunks or blocks

code/dataloaders/dataset.py:250

↓ 2 callersFunctiongrouper

Collect data into fixed-length chunks or blocks

code/dataloaders/brats2019.py:233

↓ 2 callersFunctioninitialize_network

This is specific to the U-Net and must be adapted for other network architectures :return:

code/networks/nnunet.py:500

↓ 2 callersFunctionmeasure_fp_bp_time

(model, x, y)

code/networks/networks_other.py:203

↓ 2 callersFunctionrandom_rot_flip

(image, label=None)

code/dataloaders/dataset.py:75

↓ 2 callersFunctionrefresh_policies

(db_train, cta)

code/train_fixmatch_cta.py:153

↓ 2 callersMethodresize

(self, image)

code/dataloaders/dataset.py:201

↓ 2 callersFunctiontest_single_volume_ds

(image, label, net, classes, patch_size=[256, 256])

code/val_2D.py:42

↓ 2 callersMethodupdate_rates

(self, policy, proximity)

code/augmentations/ctaugment.py:84

↓ 2 callersFunctionwindow_partition

Args: x: (B, H, W, C) window_size (int): window size Returns: windows: (num_windows*B, window_size, window_size, C)

code/networks/swin_transformer_unet_skip_expand_decoder_sys.py:28

↓ 1 callersFunctionInference

(FLAGS)

code/test_3D.py:20

↓ 1 callersFunctionInference

(FLAGS)

code/test_urpc.py:25

↓ 1 callersFunctionInference

(FLAGS)

code/test_2D_fully.py:79

↓ 1 callersMethod__init__

(self, n_classes)

code/utils/losses.py:157

↓ 1 callersMethod__init__

(self, num_classes, ndf=64, n_channel=1)

code/networks/discriminator.py:60

↓ 1 callersMethod__init__

(self)

code/networks/neural_network.py:51

↓ 1 callersMethod__init__

(self, feature_scale=4, n_classes=21, is_deconv=True, in_channels=3, nonlocal_mode='concatena

code/networks/attention_unet.py:11

↓ 1 callersMethod_dice_loss

(self, score, target)

code/utils/losses.py:169

↓ 1 callersMethod_internal_predict_3D_2Dconv

(self, x: np.ndarray, min_size: Tuple[int, int], do_mirroring: bool, mirro

code/networks/neural_network.py:806

↓ 1 callersMethod_internal_predict_3D_2Dconv_tiled

(self, x: np.ndarray, patch_size: Tuple[int, int], do_mirroring: bool,

code/networks/neural_network.py:856

↓ 1 callersMethod_internal_predict_3D_3Dconv

This one does fully convolutional inference. No sliding window

code/networks/neural_network.py:515

↓ 1 callersMethod_internal_predict_3D_3Dconv_tiled

(self, x: np.ndarray, step_size: float, do_mirroring: bool, mirror_axes: tuple,

code/networks/neural_network.py:321

↓ 1 callersMethod_one_hot_encoder

(self, input_tensor)

code/utils/losses.py:161

↓ 1 callersFunction_update_config_from_file

(config, cfg_file)

code/config.py:175

↓ 1 callersFunction_update_config_from_file

(config, cfg_file)

code/networks/config.py:175

↓ 1 callersMethodbuild_conv_block

(self, dim, padding_type, norm_layer, use_dropout, use_bias)

code/networks/networks_other.py:359

↓ 1 callersFunctioncal_metric

(gt, pred)

code/val_urpc_util.py:82

↓ 1 callersFunctioncal_metric

(gt, pred)

code/val_3D.py:82

↓ 1 callersFunctioncalculate_metric_percase

(pred, gt)

code/test_urpc_util.py:153

↓ 1 callersFunctioncalculate_metric_percase

(pred, gt)

code/test_3D_util.py:147

↓ 1 callersFunctioncolor_jitter

(image)

code/dataloaders/dataset.py:95

↓ 1 callersFunctioncreate_model

(ema=False)

code/train_cross_teaching_between_cnn_transformer_2D.py:161

↓ 1 callersFunctiondecode_segmap

Decode segmentation class labels into a color image Args: label_mask (np.ndarray): an (M,N) array of integer values denoting the

code/dataloaders/utils.py:80

↓ 1 callersFunctiondice_loss1

(score, target)

code/utils/losses.py:19

↓ 1 callersMethodfeature_based_noise

(self, x)

code/networks/unet.py:293

↓ 1 callersMethodforward_features

(self, x)

code/networks/swin_transformer_unet_skip_expand_decoder_sys.py:746

↓ 1 callersMethodforward_up_features

(self, x, x_downsample)

code/networks/swin_transformer_unet_skip_expand_decoder_sys.py:762

↓ 1 callersMethodgenerate_indexes_epoch

(self)

code/utils/util.py:142

↓ 1 callersFunctionget_cityscapes_labels

()

code/dataloaders/utils.py:19

next →1–100 of 575, ranked by callers