MCPcopy Create free account

hub / github.com/MrGiovanni/UNetPlusPlus / functions

Functions963 in github.com/MrGiovanni/UNetPlusPlus

↓ 110 callersMethodprint_to_log_file
(self, *args, also_print_to_console=True, add_timestamp=True)
pytorch/nnunet/training/network_training/network_trainer.py:224
↓ 75 callersFunctionconv2d_bn
Utility function to apply conv + BN. # Arguments x: input tensor. filters: filters in `Conv2D`. num_row: height of the con
keras/segmentation_models/backbones/inception_v3.py:45
↓ 39 callersFunctionconv2d_bn
Utility function to apply conv + BN. # Arguments x: input tensor. filters: filters in `Conv2D`. kernel_size: kernel size a
keras/segmentation_models/backbones/inception_resnet_v2.py:59
↓ 33 callersFunctionstandard_unit
(input_tensor, stage, nb_filter, kernel_size=3)
keras/helper_functions.py:77
↓ 30 callersMethodstep
(self, closure=None)
pytorch/nnunet/training/optimizer/ranger.py:68
↓ 26 callersMethod_compute_steps_for_sliding_window
(patch_size: Tuple[int, ...], image_size: Tuple[int, ...], step_size: float)
pytorch/nnunet/network_architecture/neural_network.py:262
↓ 24 callersMethodbackward
(ctx, grad_output)
pytorch/nnunet/utilities/distributed.py:39
↓ 22 callersMethodget_device
(self)
pytorch/nnunet/network_architecture/neural_network.py:32
↓ 21 callersFunctionConv2DTranspose
(filters, upsample_rate, kernel_size=(4,4), up_nam
keras/segmentation_models/linknet/blocks.py:60
↓ 20 callersMethodcompute_approx_vram_consumption
This only applies for num_conv_per_stage and convolutional_upsampling=True not real vram consumption. just a constant term to which t
pytorch/nnunet/network_architecture/generic_XNet.py:478
↓ 20 callersFunctionmaybe_to_torch
(d)
pytorch/nnunet/utilities/to_torch.py:18
↓ 20 callersFunctionto_cuda
(data, non_blocking=True, gpu_id=0)
pytorch/nnunet/utilities/to_torch.py:26
↓ 17 callersFunctionget_pool_and_conv_props
:param spacing: :param patch_size: :param min_feature_map_size: min edge length of feature maps in bottleneck :return:
pytorch/nnunet/experiment_planning/common_utils.py:89
↓ 17 callersFunctionunpack_dataset
unpacks all npz files in a folder to npy (whatever you want to have unpacked must be saved unter key) :param folder: :param threads:
pytorch/nnunet/training/dataloading/dataset_loading.py:58
↓ 16 callersMethodload_plans_file
This is what actually configures the entire experiment. The plans file is generated by experiment planning :return:
pytorch/nnunet/training/network_training/nnUNetTrainer.py:318
↓ 16 callersFunctionsum_tensor
(inp, axes, keepdim=False)
pytorch/nnunet/utilities/tensor_utilities.py:20
↓ 15 callersFunctionget_patch_size
(final_patch_size, rot_x, rot_y, rot_z, scale_range)
pytorch/nnunet/training/data_augmentation/default_data_augmentation.py:107
↓ 13 callersFunctionget_pool_and_conv_props_poolLateV2
:param spacing: :param patch_size: :param min_feature_map_size: min edge length of feature maps in bottleneck :return:
pytorch/nnunet/experiment_planning/common_utils.py:50
↓ 13 callersFunctionsave_as_nifti
(arr, filename, spacing)
pytorch/nnunet/dataset_conversion/Task061_CREMI.py:44
↓ 11 callersMethodget_matrix
(self)
pytorch/nnunet/evaluation/metrics.py:80
↓ 11 callersFunctionget_tp_fp_fn_tn
net_output must be (b, c, x, y(, z))) gt must be a label map (shape (b, 1, x, y(, z)) OR shape (b, x, y(, z))) or one hot encoding (b, c, x,
pytorch/nnunet/training/loss_functions/dice_loss.py:100
↓ 10 callersMethodget_existence
(self)
pytorch/nnunet/evaluation/metrics.py:95
↓ 10 callersMethodload_dataset
(self)
pytorch/nnunet/training/network_training/nnUNetTrainer.py:393
↓ 9 callersFunctionaggregate_scores
test = predicted image :param test_ref_pairs: :param evaluator: :param labels: must be a dict of int-> str or a list of int :para
pytorch/nnunet/evaluation/evaluator.py:321
↓ 9 callersFunctionget_lowres_axis
(new_spacing)
pytorch/nnunet/preprocessing/preprocessing.py:31
↓ 9 callersFunctionget_moreDA_augmentation
(dataloader_train, dataloader_val, patch_size, params=default_3D_augmentation_params,
pytorch/nnunet/training/data_augmentation/default_data_augmentation.py:307
↓ 9 callersMethodprocess_plans
(self, plans)
pytorch/nnunet/training/network_training/BraTS_trainer/nnUNetTrainerV2BraTSRegions.py:78
↓ 9 callersFunctionrecursive_rename
(folder)
pytorch/nnunet/utilities/recursive_rename_taskXX_to_taskXXX.py:20
↓ 9 callersFunctionto_tuple
(x)
keras/segmentation_models/utils.py:81
↓ 8 callersFunctionconvert_id_to_task_name
(task_id: int)
pytorch/nnunet/utilities/task_name_id_conversion.py:21
↓ 8 callersFunctionevaluate_regions
(folder_predicted: str, folder_gt: str, regions: dict, processes=default_num_threads)
pytorch/nnunet/evaluation/region_based_evaluation.py:53
↓ 8 callersMethodinitialize_optimizer_and_scheduler
(self)
pytorch/nnunet/training/network_training/nnUNet_variants/optimizer_and_lr/nnUNetTrainerV2_Adam.py:22
↓ 8 callersMethodload_best_checkpoint
(self, train=True)
pytorch/nnunet/training/network_training/network_trainer.py:289
↓ 8 callersMethodpredict_preprocessed_data_return_seg_and_softmax
:param data: :param do_mirroring: :param mirror_axes: :param use_sliding_window: :param step_size: :p
pytorch/nnunet/training/network_training/nnUNetTrainer.py:480
↓ 8 callersFunctionrecursive_find_python_class
(folder, trainer_name, current_module)
pytorch/nnunet/training/model_restore.py:23
↓ 8 callersMethodset_reference
Set the reference segmentation.
pytorch/nnunet/evaluation/evaluator.py:104
↓ 8 callersMethodset_test
Set the test segmentation.
pytorch/nnunet/evaluation/evaluator.py:99
↓ 7 callersFunctionConv2DBlock
Extension of Conv2D layer with batchnorm
keras/segmentation_models/common/blocks.py:6
↓ 7 callersMethod__init__
square_volumes will square the weight term. The paper recommends square_volumes=True; I don't (just an intuition)
pytorch/nnunet/training/loss_functions/dice_loss.py:26
↓ 7 callersFunctionapply_threshold_to_folder
(folder_in, folder_out, threshold, replace_with, processes=24)
pytorch/nnunet/dataset_conversion/Task082_BraTS_2020.py:80
↓ 7 callersFunctionget_layer_number
Help find layer in Keras model by name Args: model: Keras `Model` layer_name: str, name of layer Returns: index
keras/segmentation_models/utils.py:5
↓ 7 callersMethodload_latest_checkpoint
(self, train=True)
pytorch/nnunet/training/network_training/network_trainer.py:299
↓ 7 callersFunctionload_model_weights
(weights_collection, model, dataset, classes, include_top)
keras/segmentation_models/backbones/classification_models/classification_models/utils.py:11
↓ 7 callersFunctionpoly_lr
(epoch, max_epochs, initial_lr, exponent=0.9)
pytorch/nnunet/training/learning_rate/poly_lr.py:16
↓ 6 callersMethod_maybe_init_amp
(self)
pytorch/nnunet/training/network_training/network_trainer.py:440
↓ 6 callersMethoddo_split
This is a suggestion for if your dataset is a dictionary (my personal standard) :return:
pytorch/nnunet/training/network_training/network_trainer.py:149
↓ 6 callersFunctionfreeze_model
(model)
keras/segmentation_models/utils.py:69
↓ 6 callersFunctionget_backbone
(name, *args, **kwargs)
keras/segmentation_models/backbones/backbones.py:31
↓ 6 callersFunctionget_bn_params
(**params)
keras/segmentation_models/backbones/classification_models/classification_models/resnet/params.py:15
↓ 6 callersFunctionget_brats_regions
this is only valid for the brats data in here where the labels are 1, 2, and 3. The original brats data have a different labeling convention!
pytorch/nnunet/evaluation/region_based_evaluation.py:12
↓ 6 callersFunctionget_do_separate_z
(spacing, anisotropy_threshold=RESAMPLING_SEPARATE_Z_ANISO_THRESHOLD)
pytorch/nnunet/preprocessing/preprocessing.py:26
↓ 6 callersFunctionget_output_folder_name
Retrieves the correct output directory for the nnU-Net model described by the input parameters :param model: :param task: :param tra
pytorch/nnunet/utilities/folder_names.py:20
↓ 6 callersFunctionload_sample
(filename)
pytorch/nnunet/dataset_conversion/Task061_CREMI.py:30
↓ 5 callersMethod__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
pytorch/nnunet/training/network_training/BraTS_trainer/nnUNetTrainerV2BraTSRegions.py:70
↓ 5 callersFunctionbuild_resnet
TODO
keras/segmentation_models/backbones/classification_models/classification_models/resnet/builder.py:28
↓ 5 callersFunctioncopy_geometry
(image: sitk.Image, ref: sitk.Image)
pytorch/nnunet/utilities/sitk_stuff.py:19
↓ 5 callersMethodcreate_nest
(self, z, num_pool, final_num_features, num_conv_per_stage, basic_block, transpconv)
pytorch/nnunet/network_architecture/generic_UNetPlusPlus.py:435
↓ 5 callersMethodcreate_nest
(self, z, num_pool, final_num_features, num_conv_per_stage, basic_block, transpconv)
pytorch/nnunet/network_architecture/generic_XNet.py:435
↓ 5 callersFunctiondetermine_postprocessing
:param base: :param gt_labels_folder: subfolder of base with niftis of ground truth labels :param raw_subfolder_name: subfolder of base w
pytorch/nnunet/postprocessing/connected_components.py:122
↓ 5 callersFunctionforeground_mean
(filename)
pytorch/nnunet/evaluation/add_mean_dice_to_json.py:21
↓ 5 callersMethodget_basic_generators
(self)
pytorch/nnunet/training/network_training/nnUNetTrainer.py:396
↓ 5 callersMethodget_basic_generators
(self)
pytorch/nnunet/training/network_training/nnUNet_variants/data_augmentation/nnUNetTrainerV2_noDA.py:33
↓ 5 callersFunctionget_conv_params
(**params)
keras/segmentation_models/backbones/classification_models/classification_models/resnet/params.py:5
↓ 5 callersFunctionget_default_network_config
returns a dictionary that contains pointers to conv, nonlin and norm ops and the default kwargs I like to use :return:
pytorch/nnunet/network_architecture/generic_modular_UNet.py:31
↓ 5 callersMethodinitialize
For prediction of test cases just set training=False, this will prevent loading of training data and training batchgenerator initiali
pytorch/nnunet/training/network_training/nnUNetTrainer.py:189
↓ 5 callersFunctionload_postprocessing
loads the relevant part of the pkl file that is needed for applying postprocessing :param pkl_file: :return:
pytorch/nnunet/postprocessing/connected_components.py:108
↓ 5 callersFunctionresample_data_or_seg
separate_z=True will resample with order 0 along z :param data: :param new_shape: :param is_seg: :param axis: :param order:
pytorch/nnunet/preprocessing/preprocessing.py:109
↓ 5 callersFunctionresample_patient
:param cval_seg: :param cval_data: :param data: :param seg: :param original_spacing: :param target_spacing: :param order_
pytorch/nnunet/preprocessing/preprocessing.py:36
↓ 5 callersMethodrun_iteration
(self, data_generator, do_backprop=True, run_online_evaluation=False)
pytorch/nnunet/training/network_training/network_trainer.py:662
↓ 5 callersMethodrun_preprocessing
(self, num_threads)
pytorch/nnunet/experiment_planning/experiment_planner_baseline_3DUNet.py:422
↓ 4 callersFunctionInterpBlock
(level, feature_map_shape, conv_filters=512, conv_kernel_size=(1,1),
keras/segmentation_models/pspnet/blocks.py:13
↓ 4 callersMethod__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
pytorch/nnunet/training/network_training/nnUNet_variants/profiling/nnUNetTrainerV2_2epochs.py:28
↓ 4 callersMethod__init__
(self, input_channels, output_channels, kernel_size, network_props, num_blocks, first_stride=None, block=Basic
pytorch/nnunet/network_architecture/custom_modules/conv_blocks.py:215
↓ 4 callersMethodanalyze_dataset
(self, collect_intensityproperties=True)
pytorch/nnunet/experiment_planning/DatasetAnalyzer.py:233
↓ 4 callersFunctioncollect_cv_niftis
(cv_folder: str, output_folder: str, validation_folder_name: str = 'validation_raw', fol
pytorch/nnunet/postprocessing/consolidate_postprocessing.py:25
↓ 4 callersMethodcompute_approx_vram_consumption
(patch_size, base_num_features, max_num_features, num_modalities, num_
pytorch/nnunet/network_architecture/generic_modular_residual_UNet.py:340
↓ 4 callersFunctionconvert_MR_seg
(loaded_png)
pytorch/nnunet/dataset_conversion/Task037_038_Chaos_Challenge.py:45
↓ 4 callersMethodevaluate
Compute metrics for segmentations.
pytorch/nnunet/evaluation/evaluator.py:152
↓ 4 callersMethodget_basic_generators
(self)
pytorch/nnunet/training/network_training/nnUNet_variants/nnUNetTrainerNoDA.py:28
↓ 4 callersFunctionget_bn_params
(**params)
keras/segmentation_models/backbones/classification_models/classification_models/resnext/params.py:15
↓ 4 callersFunctionget_case_identifier_from_npz
(case)
pytorch/nnunet/preprocessing/cropping.py:56
↓ 4 callersFunctionget_default_configuration
(network, task, network_trainer, plans_identifier=default_plans_identifier, sear
pytorch/nnunet/run/default_configuration.py:34
↓ 4 callersFunctionhandle_block_names
(stage, block)
keras/segmentation_models/backbones/classification_models/classification_models/resnet/blocks.py:11
↓ 4 callersFunctioninception_resnet_block
Adds a Inception-ResNet block. This function builds 3 types of Inception-ResNet blocks mentioned in the paper, controlled by the `block_type`
keras/segmentation_models/backbones/inception_resnet_v2.py:97
↓ 4 callersMethodinitialize_network
This is specific to the U-Net and must be adapted for other network architectures :return:
pytorch/nnunet/training/network_training/nnUNetTrainer.py:234
↓ 4 callersMethodinitialize_network
(self)
pytorch/nnunet/training/network_training/nnUNet_variants/data_augmentation/nnUNetTrainerV2_DA3.py:326
↓ 4 callersFunctionlist_to_string
(l, delim=",")
pytorch/nnunet/evaluation/model_selection/summarize_results_with_plans.py:23
↓ 4 callersMethodload_checkpoint
(self, fname, train=True)
pytorch/nnunet/training/network_training/network_trainer.py:308
↓ 4 callersMethodload_checkpoint_ram
used for if the checkpoint is already in ram :param checkpoint: :param train: :return:
pytorch/nnunet/training/network_training/network_trainer.py:383
↓ 4 callersFunctionload_png_stack
(folder)
pytorch/nnunet/dataset_conversion/Task037_038_Chaos_Challenge.py:31
↓ 4 callersFunctionpredict_next_stage
(trainer, stage_to_be_predicted_folder)
pytorch/nnunet/training/cascade_stuff/predict_next_stage.py:46
↓ 4 callersMethodprint_to_log_file
(self, *args, also_print_to_console=True)
pytorch/nnunet/training/network_training/nnUNetTrainerV2_DDP.py:121
↓ 4 callersMethodprocess_plans
(self, plans)
pytorch/nnunet/training/network_training/nnUNetTrainer.py:325
↓ 4 callersMethodsave_checkpoint
(self, fname, save_optimizer=True)
pytorch/nnunet/training/network_training/network_trainer.py:258
↓ 4 callersFunctionsave_segmentation_nifti_from_softmax
This is a utility for writing segmentations to nifto and npz. It requires the data to have been preprocessed by GenericPreprocessor because i
pytorch/nnunet/inference/segmentation_export.py:27
↓ 4 callersMethodsetup_DA_params
(self)
pytorch/nnunet/training/network_training/nnUNet_variants/cascade/nnUNetTrainerV2CascadeFullRes_DAVariants.py:20
↓ 4 callersFunctionwrite_pngs_from_nifti
(nifti, output_folder, converter=convert_seg_to_intensity_task3)
pytorch/nnunet/dataset_conversion/Task037_038_Chaos_Challenge.py:69
↓ 3 callersFunctionConvRelu
(filters, kernel_size, use_batchnorm=False, conv_name='conv', bn_name='bn', relu_name='relu')
keras/segmentation_models/xnet/blocks.py:18
↓ 3 callersFunctionConvRelu
(filters, kernel_size, use_batchnorm=False, conv_name='conv', bn_name='bn', relu_name='relu')
keras/segmentation_models/nestnet/blocks.py:18
↓ 3 callersFunctionConvRelu
(filters, kernel_size, use_batchnorm=False, conv_name='conv', bn_name='bn', relu_name='relu')
keras/segmentation_models/unet/blocks.py:17
next →1–100 of 963, ranked by callers