MCPcopy Create free account

hub / github.com/MrGiovanni/UNetPlusPlus / functions

Functions963 in github.com/MrGiovanni/UNetPlusPlus

↓ 3 callersMethod__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
pytorch/nnunet/training/network_training/nnUNet_variants/copies/nnUNetTrainerV2_copies.py:24
↓ 3 callersMethod__init__
(self, size=None, scale_factor=None, mode='nearest', align_corners=False)
pytorch/nnunet/network_architecture/generic_UNetPlusPlus.py:155
↓ 3 callersMethod__init__
(self, size=None, scale_factor=None, mode='nearest', align_corners=False)
pytorch/nnunet/network_architecture/generic_XNet.py:155
↓ 3 callersMethod__init__
(self, input_channels, base_num_features, num_blocks_per_stage_encoder, feat_map_mul_on_downscale,
pytorch/nnunet/network_architecture/generic_modular_residual_UNet.py:315
↓ 3 callersMethod__init__
(self, size=None, scale_factor=None, mode='nearest', align_corners=False)
pytorch/nnunet/network_architecture/generic_hipp_XNet.py:155
↓ 3 callersMethod__init__
(self, size=None, scale_factor=None, mode='nearest', align_corners=False)
pytorch/nnunet/network_architecture/generic_UNet.py:155
↓ 3 callersMethod_internal_predict_2D_2Dconv
This one does fully convolutional inference. No sliding window
pytorch/nnunet/network_architecture/neural_network.py:425
↓ 3 callersFunctionapply_postprocessing_to_folder
applies removing of all but the largest connected component to all niftis in a folder :param min_valid_object_size: :param min_valid_obje
pytorch/nnunet/postprocessing/connected_components.py:400
↓ 3 callersMethodcompute
(self)
pytorch/nnunet/evaluation/metrics.py:63
↓ 3 callersFunctioncompute_BraTS_HD95
ref and gt are binary integer numpy.ndarray s spacing is assumed to be (1, 1, 1) :param ref: :param pred: :return:
pytorch/nnunet/dataset_conversion/Task082_BraTS_2020.py:329
↓ 3 callersFunctioncompute_BraTS_dice
ref and gt are binary integer numpy.ndarray s :param ref: :param gt: :return:
pytorch/nnunet/dataset_conversion/Task082_BraTS_2020.py:301
↓ 3 callersFunctionconsolidate_folds
Used to determine the postprocessing for an experiment after all five folds have been completed. In the validation of each fold, the postproc
pytorch/nnunet/postprocessing/consolidate_postprocessing.py:42
↓ 3 callersMethodconstruct_labels
Construct label set from unique entries in segmentations.
pytorch/nnunet/evaluation/evaluator.py:125
↓ 3 callersFunctionconvert_labels_back_to_BraTS_2018_2019_convention
reads all prediction files (nifti) in the input folder, converts the labels back to BraTS convention and saves the result in output_folder
pytorch/nnunet/dataset_conversion/Task032_BraTS_2018.py:34
↓ 3 callersFunctioncopy_BraTS_segmentation_and_convert_labels
(in_file, out_file)
pytorch/nnunet/dataset_conversion/Task043_BraTS_2019.py:25
↓ 3 callersMethodcreate_nest
(self, z, num_pool, final_num_features, num_conv_per_stage, basic_block, transpconv)
pytorch/nnunet/network_architecture/generic_hipp_XNet.py:404
↓ 3 callersFunctioncrop
(task_string, override=False, num_threads=default_num_threads)
pytorch/nnunet/experiment_planning/utils.py:122
↓ 3 callersFunctioncrop_to_bbox
(image, bbox)
pytorch/nnunet/preprocessing/cropping.py:45
↓ 3 callersFunctioncycle_lr
(current_epoch, cycle_length=100, min_lr=1e-6, max_lr=1e-3)
pytorch/nnunet/training/network_training/nnUNet_variants/optimizer_and_lr/nnUNetTrainerV2_cycleAtEnd.py:21
↓ 3 callersFunctionevaluate_verse_folder
(folder_pred, folder_gt, out_json="/home/fabian/verse.json")
pytorch/nnunet/dataset_conversion/Task056_VerSe2019.py:66
↓ 3 callersMethodfind_lr
stolen and adapted from here: https://sgugger.github.io/how-do-you-find-a-good-learning-rate.html :param num_iters: :param in
pytorch/nnunet/training/network_training/network_trainer.py:722
↓ 3 callersFunctionget_available_models
()
pytorch/nnunet/inference/pretrained_models/download_pretrained_model.py:27
↓ 3 callersFunctionget_conv_params
(**params)
keras/segmentation_models/backbones/classification_models/classification_models/resnext/params.py:5
↓ 3 callersFunctionget_default_augmentation
(dataloader_train, dataloader_val, patch_size, params=default_3D_augmentation_params,
pytorch/nnunet/training/data_augmentation/default_data_augmentation.py:130
↓ 3 callersFunctionget_shape_must_be_divisible_by
(net_numpool_per_axis)
pytorch/nnunet/experiment_planning/common_utils.py:232
↓ 3 callersMethodinitialize
create self.output_folder modify self.output_folder if you are doing cross-validation (one folder per fold) set self.tr_gen
pytorch/nnunet/training/network_training/network_trainer.py:130
↓ 3 callersMethodinitialize_network
inference_apply_nonlin to sigmoid
pytorch/nnunet/training/network_training/BraTS_trainer/nnUNetTrainerV2BraTSRegions.py:85
↓ 3 callersMethodinitialize_optimizer_and_scheduler
(self)
pytorch/nnunet/training/network_training/nnUNetTrainer.py:267
↓ 3 callersFunctionint_shape
Returns the shape of tensor or variable as a tuple of int or None entries. # Arguments x: Tensor or variable. # Returns A tupl
keras/segmentation_models/common/functions.py:52
↓ 3 callersFunctionload_csv_for_ranking
(csv_file: str)
pytorch/nnunet/dataset_conversion/Task082_BraTS_2020.py:409
↓ 3 callersFunctionload_dataset
(folder, num_cases_properties_loading_threshold=1000)
pytorch/nnunet/training/dataloading/dataset_loading.py:89
↓ 3 callersFunctionload_model_and_checkpoint_files
used for if you need to ensemble the five models of a cross-validation. This will restore the model from the checkpoint in fold 0, load all p
pytorch/nnunet/training/model_restore.py:109
↓ 3 callersMethodmaybe_update_lr
(self)
pytorch/nnunet/training/network_training/network_trainer.py:539
↓ 3 callersFunctionmerge
(folders, output_folder, threads, override=True, postprocessing_file=None, store_npz=False)
pytorch/nnunet/inference/ensemble_predictions.py:56
↓ 3 callersFunctionpad_shape
pads shape so that it is divisibly by must_be_divisible_by :param shape: :param must_be_divisible_by: :return:
pytorch/nnunet/experiment_planning/common_utils.py:236
↓ 3 callersMethodplan_experiment
(self)
pytorch/nnunet/experiment_planning/experiment_planner_baseline_3DUNet.py:247
↓ 3 callersFunctionpredict_from_folder
here we use the standard naming scheme to generate list_of_lists and output_files needed by predict_cases :param model: :param input
pytorch/nnunet/inference/predict.py:573
↓ 3 callersFunctionpreprocess_multithreaded
(trainer, list_of_lists, output_files, num_processes=2, segs_from_prev_stage=None)
pytorch/nnunet/inference/predict.py:93
↓ 3 callersFunctionrank_algorithms
data is (metrics x experiments x cases) :param data: :return:
pytorch/nnunet/dataset_conversion/Task082_BraTS_2020.py:419
↓ 3 callersFunctionrestore_model
This is a utility function to load any nnUNet trainer from a pkl. It will recursively search nnunet.trainig.network_training for the file tha
pytorch/nnunet/training/model_restore.py:44
↓ 3 callersMethodrun_training
(self)
pytorch/nnunet/training/network_training/nnUNetTrainer.py:299
↓ 3 callersMethodsetup_DA_params
(self)
pytorch/nnunet/training/network_training/BraTS_trainer/nnUNetTrainerV2BraTSRegions.py:530
↓ 3 callersFunctionsummarize2
(task_ids, models=('2d', '3d_lowres', '3d_fullres', '3d_cascade_fullres'), output_dir=join(netw
pytorch/nnunet/evaluation/model_selection/summarize_results_in_one_json.py:101
↓ 3 callersFunctionto_one_hot
(seg, all_seg_labels=None)
pytorch/nnunet/utilities/one_hot_encoding.py:18
↓ 3 callersMethodvalidate
if debug=True then the temporary files generated for postprocessing determination will be kept
pytorch/nnunet/training/network_training/nnUNetTrainer.py:520
↓ 2 callersFunctionConvRelu
(filters, kernel_size, use_batchnorm=False, conv_name='conv',
keras/segmentation_models/linknet/blocks.py:18
↓ 2 callersFunctionGroupConv2D
(filters, kernel_size, conv_params, conv_name, strides=(1,1), cardinality=32)
keras/segmentation_models/backbones/classification_models/classification_models/resnext/blocks.py:22
↓ 2 callersFunctionNestnet
Args: backbone_name: (str) look at list of available backbones. input_shape: (tuple) dimensions of input data (H, W, C)
keras/segmentation_models/nestnet/model.py:42
↓ 2 callersFunctionUnet
Args: backbone_name: (str) look at list of available backbones. input_shape: (tuple) dimensions of input data (H, W, C)
keras/segmentation_models/unet/model.py:24
↓ 2 callersFunctionXnet
Args: backbone_name: (str) look at list of available backbones. input_shape: (tuple) dimensions of input data (H, W, C)
keras/segmentation_models/xnet/model.py:42
↓ 2 callersMethod__init__
(self, input_channels, base_num_features, num_blocks_per_stage_encoder, feat_map_mul_on_downscale,
pytorch/nnunet/network_architecture/generic_modular_UNet.py:321
↓ 2 callersMethod_get_gaussian
(patch_size, sigma_scale=1. / 8)
pytorch/nnunet/network_architecture/neural_network.py:246
↓ 2 callersMethod_internal_maybe_mirror_and_pred_2D
(self, x: Union[np.ndarray, torch.tensor], mirror_axes: tuple, do_m
pytorch/nnunet/network_architecture/neural_network.py:559
↓ 2 callersMethod_internal_maybe_mirror_and_pred_3D
(self, x: Union[np.ndarray, torch.tensor], mirror_axes: tuple, do_m
pytorch/nnunet/network_architecture/neural_network.py:501
↓ 2 callersMethod_internal_predict_2D_2Dconv_tiled
(self, x: np.ndarray, step_size: float, do_mirroring: bool, mirror_axes: tuple,
pytorch/nnunet/network_architecture/neural_network.py:602
↓ 2 callersFunctionanalyze_dataset
(task_string, override=False, collect_intensityproperties=True, num_processes=default_num_threads)
pytorch/nnunet/experiment_planning/utils.py:138
↓ 2 callersFunctionbuild_resnext
TODO
keras/segmentation_models/backbones/classification_models/classification_models/resnext/builder.py:29
↓ 2 callersFunctioncheck_if_valid
(ensemble: str, valid_models, valid_trainers, valid_plans)
pytorch/nnunet/inference/pretrained_models/collect_pretrained_models.py:79
↓ 2 callersMethodcompute_approx_vram_consumption
(patch_size, base_num_features, max_num_features, num_modalities, num_
pytorch/nnunet/network_architecture/generic_modular_UNet.py:341
↓ 2 callersMethodcompute_loss
(self, output, target)
pytorch/nnunet/training/network_training/nnUNetTrainerV2_DDP.py:259
↓ 2 callersMethodcompute_loss
(self, ces, tps, fps, fns)
pytorch/nnunet/training/network_training/nnUNetTrainerV2_DP.py:228
↓ 2 callersFunctionconv_block
The identity block is the block that has no conv layer at shortcut. # Arguments input_tensor: input tensor kernel_size: default 3,
keras/segmentation_models/backbones/classification_models/classification_models/resnet/blocks.py:89
↓ 2 callersFunctionconv_block
The conv block is the block that has conv layer at shortcut. # Arguments filters: integer, used for first and second conv layers, third co
keras/segmentation_models/backbones/classification_models/classification_models/resnext/blocks.py:41
↓ 2 callersFunctioncopy_ensembles
(taskname, output_folder, valid_models=('2d', '3d_fullres', '3d_lowres', '3d_cascade_fullres'),
pytorch/nnunet/inference/pretrained_models/collect_pretrained_models.py:93
↓ 2 callersFunctioncopy_pretrained_models_for_task
(task_name: str, output_directory: str, models: tuple = ("2d", "3d_lowres"
pytorch/nnunet/inference/pretrained_models/collect_pretrained_models.py:57
↓ 2 callersFunctioncreate_lists_from_splitted_dataset
(base_folder_splitted)
pytorch/nnunet/experiment_planning/utils.py:82
↓ 2 callersFunctioncreate_region_from_mask
(mask, join_labels: tuple)
pytorch/nnunet/evaluation/region_based_evaluation.py:34
↓ 2 callersMethodcrop_from_list_of_files
(data_files, seg_file=None)
pytorch/nnunet/preprocessing/cropping.py:153
↓ 2 callersMethoddetermine_normalization_scheme
(self)
pytorch/nnunet/experiment_planning/experiment_planner_baseline_3DUNet.py:359
↓ 2 callersMethoddetermine_whether_to_use_mask_for_norm
(self)
pytorch/nnunet/experiment_planning/experiment_planner_baseline_3DUNet.py:380
↓ 2 callersFunctiondice_coef
(y_true, y_pred)
keras/helper_functions.py:37
↓ 2 callersMethoddo_split
(self)
pytorch/nnunet/training/network_training/nnUNetTrainerCascadeFullRes.py:60
↓ 2 callersMethoddo_split
(self)
pytorch/nnunet/training/network_training/nnUNetTrainerV2_CascadeFullRes.py:60
↓ 2 callersFunctiondownload_and_install_from_url
(url)
pytorch/nnunet/inference/pretrained_models/download_pretrained_model.py:208
↓ 2 callersFunctionensemble
(training_output_folder1, training_output_folder2, output_folder, task, validation_folder, folds)
pytorch/nnunet/evaluation/model_selection/ensemble.py:39
↓ 2 callersFunctionevaluate_BraTS_folder
(folder_pred, folder_gt, num_processes: int = 24, strict=False)
pytorch/nnunet/dataset_conversion/Task082_BraTS_2020.py:389
↓ 2 callersFunctionextract_outputs
Help extract intermediate layer outputs from model Args: model: Keras `Model` layer: list of integers/str, list of layers ind
keras/segmentation_models/utils.py:24
↓ 2 callersFunctionget_case_identifiers
(folder)
pytorch/nnunet/training/dataloading/dataset_loading.py:26
↓ 2 callersMethodget_classes
(self)
pytorch/nnunet/experiment_planning/DatasetAnalyzer.py:109
↓ 2 callersFunctionget_insaneDA_augmentation2
(dataloader_train, dataloader_val, patch_size, params=default_3D_augmentation_params,
pytorch/nnunet/training/network_training/nnUNet_variants/data_augmentation/nnUNetTrainerV2_DA3.py:41
↓ 2 callersFunctionget_mean_foreground_dice
(json_file)
pytorch/nnunet/evaluation/model_selection/figure_out_what_to_submit.py:36
↓ 2 callersFunctionget_no_augmentation
use this instead of get_default_augmentation (drop in replacement) to turn off all data augmentation :param dataloader_train: :param data
pytorch/nnunet/training/data_augmentation/default_data_augmentation.py:231
↓ 2 callersFunctionget_patient_identifiers_from_cropped_files
(folder)
pytorch/nnunet/preprocessing/cropping.py:119
↓ 2 callersMethodget_properties_for_stage
Computation of input patch size starts out with the new median shape (in voxels) of a dataset. This is opposed to prior experiments w
pytorch/nnunet/experiment_planning/experiment_planner_baseline_3DUNet.py:144
↓ 2 callersMethodget_target_spacing
(self)
pytorch/nnunet/experiment_planning/experiment_planner_baseline_3DUNet.py:66
↓ 2 callersFunctionhandle_block_names
(stage, block)
keras/segmentation_models/backbones/classification_models/classification_models/resnext/blocks.py:13
↓ 2 callersFunctionhandle_block_names
(stage, cols)
keras/segmentation_models/xnet/blocks.py:9
↓ 2 callersFunctionhandle_block_names
(stage, cols)
keras/segmentation_models/nestnet/blocks.py:9
↓ 2 callersFunctionhandle_block_names
(stage)
keras/segmentation_models/unet/blocks.py:9
↓ 2 callersFunctioninstall_model_from_zip_file
(zip_file: str)
pytorch/nnunet/inference/pretrained_models/download_pretrained_model.py:255
↓ 2 callersFunctionload_corr_save
(in_folder: str, out_folder: str, filename: str)
pytorch/nnunet/dataset_conversion/Task056_VerSe2019.py:30
↓ 2 callersMethodload_properties_of_cropped
(self, case_identifier)
pytorch/nnunet/experiment_planning/experiment_planner_baseline_3DUNet.py:375
↓ 2 callersMethodload_properties_of_cropped
(self, case_identifier)
pytorch/nnunet/experiment_planning/DatasetAnalyzer.py:45
↓ 2 callersFunctionpermute_dimensions
Permutes axes in a tensor. # Arguments x: Tensor or variable. pattern: A tuple of dimension indices, e.g. `(0, 2, 1)`.
keras/segmentation_models/common/functions.py:40
↓ 2 callersFunctionplan_and_preprocess
(task_string, processes_lowres=default_num_threads, processes_fullres=3, no_preprocessing=False)
pytorch/nnunet/experiment_planning/utils.py:144
↓ 2 callersMethodplan_experiment
(self)
pytorch/nnunet/experiment_planning/experiment_planner_baseline_2DUNet.py:90
↓ 2 callersFunctionprecision
TP / (TP + FP)
pytorch/nnunet/evaluation/metrics.py:141
↓ 2 callersMethodpredict_3D
Use this function to predict a 3D image. It does not matter whether the network is a 2D or 3D U-Net, it will detect that automaticall
pytorch/nnunet/network_architecture/neural_network.py:73
↓ 2 callersMethodpredict_preprocessed_data_return_seg_and_softmax
(self, data: np.ndarray, do_mirroring: bool = True, m
pytorch/nnunet/training/network_training/nnUNet_variants/profiling/nnUNetTrainerV2_2epochs.py:40
↓ 2 callersMethodpreprocess_patient
Used to predict new unseen data. Not used for the preprocessing of the training/test data :param input_files: :return:
pytorch/nnunet/training/network_training/nnUNetTrainer.py:416
← previousnext →101–200 of 963, ranked by callers