MCPcopy Create free account

hub / github.com/YaoZhang93/MAML / functions

Functions895 in github.com/YaoZhang93/MAML

↓ 134 callersMethodprint_to_log_file
(self, *args, also_print_to_console=True, add_timestamp=True)
nnunet/training/network_training/network_trainer.py:224
↓ 35 callersMethodstep
(self, closure=None)
nnunet/training/optimizer/ranger.py:68
↓ 28 callersMethodbackward
(ctx, grad_output)
nnunet/utilities/distributed.py:39
↓ 23 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
nnunet/network_architecture/generic_UNet.py:411
↓ 22 callersFunctionget_pool_and_conv_props
:param spacing: :param patch_size: :param min_feature_map_size: min edge length of feature maps in bottleneck :return:
nnunet/experiment_planning/common_utils.py:89
↓ 22 callersFunctionmaybe_to_torch
(d)
nnunet/utilities/to_torch.py:18
↓ 22 callersFunctionto_cuda
(data, non_blocking=True, gpu_id=0)
nnunet/utilities/to_torch.py:26
↓ 18 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:
nnunet/training/dataloading/dataset_loading.py:58
↓ 17 callersFunctionget_patch_size
(final_patch_size, rot_x, rot_y, rot_z, scale_range)
nnunet/training/data_augmentation/default_data_augmentation.py:111
↓ 17 callersMethodload_plans_file
This is what actually configures the entire experiment. The plans file is generated by experiment planning :return:
nnunet/training/network_training/nnUNetTrainer.py:319
↓ 17 callersFunctionremove_label
(label_file: str, remove_this: int, replace_with: int = 0)
nnunet/dataset_conversion/Task056_VerSe2019.py:107
↓ 16 callersMethodget_device
(self)
nnunet/network_architecture/neural_network.py:32
↓ 16 callersFunctionsum_tensor
(inp, axes, keepdim=False)
nnunet/utilities/tensor_utilities.py:20
↓ 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:
nnunet/experiment_planning/common_utils.py:50
↓ 13 callersMethodprint_to_log_file
(self, *args, also_print_to_console=True)
nnunet/training/network_training/nnUNetTrainerV2_DDP.py:126
↓ 13 callersFunctionsave_as_nifti
(arr, filename, spacing)
nnunet/dataset_conversion/Task061_CREMI.py:44
↓ 11 callersFunctionconvert_id_to_task_name
(task_id: int)
nnunet/utilities/task_name_id_conversion.py:21
↓ 11 callersMethodget_matrix
(self)
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,
nnunet/training/loss_functions/dice_loss.py:100
↓ 11 callersMethodload_dataset
(self)
nnunet/training/network_training/nnUNetTrainer.py:394
↓ 10 callersMethod_compute_steps_for_sliding_window
(patch_size: Tuple[int, ...], image_size: Tuple[int, ...], step_size: float)
nnunet/network_architecture/neural_network.py:262
↓ 10 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
nnunet/evaluation/evaluator.py:321
↓ 10 callersMethodget_existence
(self)
nnunet/evaluation/metrics.py:95
↓ 10 callersFunctionget_moreDA_augmentation
(dataloader_train, dataloader_val, patch_size, params=default_3D_augmentation_params,
nnunet/training/data_augmentation/data_augmentation_moreDA.py:41
↓ 9 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:
nnunet/network_architecture/generic_modular_UNet.py:31
↓ 9 callersFunctionget_lowres_axis
(new_spacing)
nnunet/preprocessing/preprocessing.py:33
↓ 9 callersFunctionrecursive_rename
(folder)
nnunet/utilities/recursive_rename_taskXX_to_taskXXX.py:20
↓ 8 callersFunctionevaluate_regions
(folder_predicted: str, folder_gt: str, regions: dict, processes=default_num_threads)
nnunet/evaluation/region_based_evaluation.py:53
↓ 8 callersMethodinitialize_optimizer_and_scheduler
(self)
nnunet/training/network_training/nnUNet_variants/optimizer_and_lr/nnUNetTrainerV2_Adam.py:22
↓ 8 callersFunctionpoly_lr
(epoch, max_epochs, initial_lr, exponent=0.9)
nnunet/training/learning_rate/poly_lr.py:16
↓ 8 callersMethodprocess_plans
(self, plans)
nnunet/training/network_training/competitions_with_custom_Trainers/BraTS2020/nnUNetTrainerV2BraTSRegions.py:75
↓ 8 callersFunctionrecursive_find_python_class
(folder, trainer_name, current_module)
nnunet/training/model_restore.py:23
↓ 8 callersFunctionresample_patient
:param data: :param seg: :param original_spacing: :param target_spacing: :param order_data: :param order_seg: :param forc
nnunet/preprocessing/preprocessing.py:38
↓ 8 callersMethodset_reference
Set the reference segmentation.
nnunet/evaluation/evaluator.py:104
↓ 8 callersMethodset_test
Set the test segmentation.
nnunet/evaluation/evaluator.py:99
↓ 7 callersMethod__init__
square_volumes will square the weight term. The paper recommends square_volumes=True; I don't (just an intuition)
nnunet/training/loss_functions/dice_loss.py:26
↓ 7 callersMethod_maybe_init_amp
(self)
nnunet/training/network_training/network_trainer.py:402
↓ 7 callersFunctionapply_threshold_to_folder
(folder_in, folder_out, threshold, replace_with, processes=24)
nnunet/dataset_conversion/Task082_BraTS_2020.py:75
↓ 7 callersMethoddo_split
The default split is a 5 fold CV on all available training cases. nnU-Net will create a split (it is seeded, so always the same) and
nnunet/training/network_training/MAMLTrainerV2.py:288
↓ 7 callersMethodpredict_preprocessed_data_return_seg_and_softmax
We need to wrap this because we need to enforce self.network.do_ds = False for prediction
nnunet/training/network_training/MAMLTrainerV2.py:207
↓ 6 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
nnunet/postprocessing/connected_components.py:122
↓ 6 callersMethodget_basic_generators
(self)
nnunet/training/network_training/nnUNetTrainer.py:397
↓ 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!
nnunet/evaluation/region_based_evaluation.py:12
↓ 6 callersFunctionget_do_separate_z
(spacing, anisotropy_threshold=RESAMPLING_SEPARATE_Z_ANISO_THRESHOLD)
nnunet/preprocessing/preprocessing.py:28
↓ 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
nnunet/utilities/folder_names.py:20
↓ 6 callersFunctionload_sample
(filename)
nnunet/dataset_conversion/Task061_CREMI.py:30
↓ 6 callersFunctionreorient_all_images_in_folder_to_ras
(folder: str, num_processes: int = 8)
nnunet/utilities/image_reorientation.py:68
↓ 5 callersFunctioncollect_cv_niftis
(cv_folder: str, output_folder: str, validation_folder_name: str = 'validation_raw', fol
nnunet/postprocessing/consolidate_postprocessing.py:25
↓ 5 callersFunctioncopy_geometry
(image: sitk.Image, ref: sitk.Image)
nnunet/utilities/sitk_stuff.py:19
↓ 5 callersFunctionforeground_mean
(filename)
nnunet/evaluation/add_mean_dice_to_json.py:21
↓ 5 callersMethodget_basic_generators
(self)
nnunet/training/network_training/nnUNet_variants/data_augmentation/nnUNetTrainerV2_noDA.py:33
↓ 5 callersMethodinitialize
- replaced get_default_augmentation with get_moreDA_augmentation - enforce to only run this code once - loss function wrapper
nnunet/training/network_training/MAMLTrainerV2.py:55
↓ 5 callersMethodload_best_checkpoint
(self, train=True)
nnunet/training/network_training/network_trainer.py:290
↓ 5 callersMethodload_checkpoint
(self, fname, train=True)
nnunet/training/network_training/network_trainer.py:315
↓ 5 callersMethodload_checkpoint_ram
used for if the checkpoint is already in ram :param checkpoint: :param train: :return:
nnunet/training/network_training/network_trainer.py:339
↓ 5 callersFunctionload_postprocessing
loads the relevant part of the pkl file that is needed for applying postprocessing :param pkl_file: :return:
nnunet/postprocessing/connected_components.py:108
↓ 5 callersMethodprocess_plans
(self, plans)
nnunet/training/network_training/nnUNetTrainer.py:326
↓ 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:
nnunet/preprocessing/preprocessing.py:109
↓ 5 callersMethodrun_iteration
(self, data_generator, do_backprop=True, run_online_evaluation=False)
nnunet/training/network_training/network_trainer.py:627
↓ 5 callersMethodrun_preprocessing
(self, num_threads)
nnunet/experiment_planning/experiment_planner_baseline_3DUNet.py:424
↓ 5 callersMethodsave_my_plans
(self)
nnunet/experiment_planning/experiment_planner_baseline_3DUNet.py:81
↓ 4 callersMethod__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
nnunet/training/network_training/nnUNet_variants/benchmarking/nnUNetTrainerV2_2epochs.py:28
↓ 4 callersMethod__init__
(self, size=None, scale_factor=None, mode='nearest', align_corners=False)
nnunet/network_architecture/generic_MAML.py:155
↓ 4 callersMethod__init__
(self, input_channels, output_channels, kernel_size, network_props, num_blocks, first_stride=None, block=Basic
nnunet/network_architecture/custom_modules/conv_blocks.py:215
↓ 4 callersMethodcompute_approx_vram_consumption
(patch_size, base_num_features, max_num_features, num_modalities, num_
nnunet/network_architecture/generic_modular_residual_UNet.py:340
↓ 4 callersMethodcompute_loss
(self, ces, tps, fps, fns)
nnunet/training/network_training/nnUNetTrainerV2_DP.py:227
↓ 4 callersFunctionconvert_2d_image_to_nifti
Reads an image (must be a format that it recognized by skimage.io.imread) and converts it into a series of niftis. The image can have an arbi
nnunet/utilities/file_conversions.py:8
↓ 4 callersFunctionconvert_MR_seg
(loaded_png)
nnunet/dataset_conversion/Task037_038_Chaos_Challenge.py:45
↓ 4 callersMethoddetermine_whether_to_use_mask_for_norm
(self)
nnunet/experiment_planning/experiment_planner_baseline_3DUNet.py:382
↓ 4 callersFunctiongenerate_filename_for_nnunet
(pat_id, ts, pat_folder=None, add_zeros=False, vendor=None, centre=None, mode='mnms',
nnunet/dataset_conversion/Task114_heart_MNMs.py:40
↓ 4 callersMethodget_basic_generators
(self)
nnunet/training/network_training/nnUNet_variants/nnUNetTrainerNoDA.py:28
↓ 4 callersFunctionget_case_identifier_from_npz
(case)
nnunet/preprocessing/cropping.py:56
↓ 4 callersFunctionget_default_configuration
(network, task, network_trainer, plans_identifier=default_plans_identifier, sear
nnunet/run/default_configuration.py:34
↓ 4 callersMethodinitialize_network
- momentum 0.99 - SGD instead of Adam - self.lr_scheduler = None because we do poly_lr - deep supervision = True
nnunet/training/network_training/MAMLTrainerV2.py:136
↓ 4 callersMethodinitialize_network
(self)
nnunet/training/network_training/nnUNet_variants/data_augmentation/nnUNetTrainerV2_DA3.py:167
↓ 4 callersFunctionlist_to_string
(l, delim=",")
nnunet/evaluation/model_selection/summarize_results_with_plans.py:23
↓ 4 callersMethodload_latest_checkpoint
(self, train=True)
nnunet/training/network_training/network_trainer.py:300
↓ 4 callersFunctionload_png_stack
(folder)
nnunet/dataset_conversion/Task037_038_Chaos_Challenge.py:31
↓ 4 callersFunctionpredict_next_stage
(trainer, stage_to_be_predicted_folder)
nnunet/training/cascade_stuff/predict_next_stage.py:46
↓ 4 callersMethodrun_iteration
(self, data_generator, do_backprop=True, run_online_evaluation=False)
nnunet/training/network_training/nnUNetTrainerV2_DDP.py:204
↓ 4 callersMethodsave_checkpoint
(self, fname, save_optimizer=True)
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
nnunet/inference/segmentation_export.py:27
↓ 4 callersMethodsetup_DA_params
(self)
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)
nnunet/dataset_conversion/Task037_038_Chaos_Challenge.py:69
↓ 3 callersMethod__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
nnunet/training/network_training/nnUNet_variants/copies/nnUNetTrainerV2_copies.py:24
↓ 3 callersMethod__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
nnunet/training/network_training/competitions_with_custom_Trainers/BraTS2020/nnUNetTrainerV2BraTSRegions_moreDA.py:37
↓ 3 callersMethod__init__
(self, input_channels, base_num_features, num_blocks_per_stage_encoder, feat_map_mul_on_downscale,
nnunet/network_architecture/generic_modular_residual_UNet.py:315
↓ 3 callersMethod__init__
(self, size=None, scale_factor=None, mode='nearest', align_corners=False)
nnunet/network_architecture/generic_UNet.py:155
↓ 3 callersMethod_internal_predict_2D_2Dconv
This one does fully convolutional inference. No sliding window
nnunet/network_architecture/neural_network.py:427
↓ 3 callersMethodanalyze_dataset
(self, collect_intensityproperties=True)
nnunet/experiment_planning/DatasetAnalyzer.py:225
↓ 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
nnunet/postprocessing/connected_components.py:400
↓ 3 callersMethodcompute
(self)
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:
nnunet/dataset_conversion/Task082_BraTS_2020.py:325
↓ 3 callersFunctioncompute_BraTS_dice
ref and gt are binary integer numpy.ndarray s :param ref: :param gt: :return:
nnunet/dataset_conversion/Task082_BraTS_2020.py:296
↓ 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
nnunet/postprocessing/consolidate_postprocessing.py:43
↓ 3 callersMethodconstruct_labels
Construct label set from unique entries in segmentations.
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
nnunet/dataset_conversion/Task032_BraTS_2018.py:45
↓ 3 callersFunctioncopy_BraTS_segmentation_and_convert_labels
(in_file, out_file)
nnunet/dataset_conversion/Task043_BraTS_2019.py:25
↓ 3 callersFunctioncrop
(task_string, override=False, num_threads=default_num_threads)
nnunet/experiment_planning/utils.py:122
↓ 3 callersFunctioncrop_to_bbox
(image, bbox)
nnunet/preprocessing/cropping.py:45
next →1–100 of 895, ranked by callers