MCPcopy Create free account

hub / github.com/YaoZhang93/MAML / functions

Functions895 in github.com/YaoZhang93/MAML

↓ 3 callersFunctioncycle_lr
(current_epoch, cycle_length=100, min_lr=1e-6, max_lr=1e-3)
nnunet/training/network_training/nnUNet_variants/optimizer_and_lr/nnUNetTrainerV2_cycleAtEnd.py:21
↓ 3 callersFunctiondownload_and_install_from_url
(url)
nnunet/inference/pretrained_models/download_pretrained_model.py:252
↓ 3 callersMethodevaluate
Compute metrics for segmentations.
nnunet/evaluation/evaluator.py:152
↓ 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
nnunet/training/network_training/network_trainer.py:687
↓ 3 callersFunctiongenerate_dataset_json
:param output_file: This needs to be the full path to the dataset.json you intend to write, so output_file='DATASET_PATH/dataset.json' where
nnunet/dataset_conversion/utils.py:27
↓ 3 callersFunctionget_available_models
()
nnunet/inference/pretrained_models/download_pretrained_model.py:23
↓ 3 callersFunctionget_default_augmentation
(dataloader_train, dataloader_val, patch_size, params=default_3D_augmentation_params,
nnunet/training/data_augmentation/default_data_augmentation.py:134
↓ 3 callersFunctionget_shape_must_be_divisible_by
(net_numpool_per_axis)
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
nnunet/training/network_training/network_trainer.py:130
↓ 3 callersMethodinitialize_optimizer_and_scheduler
(self)
nnunet/training/network_training/MAMLTrainerV2.py:171
↓ 3 callersFunctionload_csv_for_ranking
(csv_file: str)
nnunet/dataset_conversion/Task082_BraTS_2020.py:405
↓ 3 callersFunctionload_dataset
(folder, num_cases_properties_loading_threshold=1000)
nnunet/training/dataloading/dataset_loading.py:89
↓ 3 callersMethodload_final_checkpoint
(self, train=False)
nnunet/training/network_training/network_trainer.py:309
↓ 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
nnunet/training/model_restore.py:109
↓ 3 callersFunctionload_pretrained_weights
THIS DOES NOT TRANSFER SEGMENTATION HEADS!
nnunet/run/load_pretrained_weights.py:17
↓ 3 callersMethodmaybe_update_lr
if epoch is not None we overwrite epoch. Else we use epoch = self.epoch + 1 (maybe_update_lr is called in on_epoch_end which is call
nnunet/training/network_training/MAMLTrainerV2.py:405
↓ 3 callersFunctionmerge
(folders, output_folder, threads, override=True, postprocessing_file=None, store_npz=False)
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:
nnunet/experiment_planning/common_utils.py:236
↓ 3 callersMethodplan_experiment
(self)
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
nnunet/inference/predict.py:604
↓ 3 callersFunctionpreprocess_multithreaded
(trainer, list_of_lists, output_files, num_processes=2, segs_from_prev_stage=None)
nnunet/inference/predict.py:93
↓ 3 callersFunctionrank_algorithms
data is (metrics x experiments x cases) :param data: :return:
nnunet/dataset_conversion/Task082_BraTS_2020.py:415
↓ 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
nnunet/training/model_restore.py:44
↓ 3 callersMethodrun_training
if we run with -c then we need to set the correct lr for the first epoch, otherwise it will run the first continued epoch with self.i
nnunet/training/network_training/MAMLTrainerV2.py:442
↓ 3 callersMethodsetup_DA_params
(self)
nnunet/training/network_training/competitions_with_custom_Trainers/BraTS2020/nnUNetTrainerV2BraTSRegions_moreDA.py:203
↓ 3 callersFunctionsummarize2
(task_ids, models=('2d', '3d_lowres', '3d_fullres', '3d_cascade_fullres'), output_dir=join(netw
nnunet/evaluation/model_selection/summarize_results_in_one_json.py:101
↓ 3 callersFunctionto_one_hot
(seg, all_seg_labels=None)
nnunet/utilities/one_hot_encoding.py:18
↓ 3 callersMethodvalidate
We need to wrap this because we need to enforce self.network.do_ds = False for prediction
nnunet/training/network_training/MAMLTrainerV2.py:189
↓ 2 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.py:67
↓ 2 callersMethod__init__
(self, input_channels, base_num_features, num_blocks_per_stage_encoder, feat_map_mul_on_downscale,
nnunet/network_architecture/generic_modular_UNet.py:322
↓ 2 callersMethod_get_gaussian
(patch_size, sigma_scale=1. / 8)
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
nnunet/network_architecture/neural_network.py:566
↓ 2 callersMethod_internal_maybe_mirror_and_pred_3D
(self, x: Union[np.ndarray, torch.tensor], mirror_axes: tuple, do_m
nnunet/network_architecture/neural_network.py:499
↓ 2 callersMethod_internal_predict_2D_2Dconv_tiled
(self, x: np.ndarray, step_size: float, do_mirroring: bool, mirror_axes: tuple,
nnunet/network_architecture/neural_network.py:616
↓ 2 callersFunctionanalyze_dataset
(task_string, override=False, collect_intensityproperties=True, num_processes=default_num_threads)
nnunet/experiment_planning/utils.py:138
↓ 2 callersFunctioncheck_if_all_in_good_orientation
(imagesTr_folder: str, labelsTr_folder: str, output_folder: str)
nnunet/dataset_conversion/Task056_VerSe2019.py:28
↓ 2 callersFunctioncheck_if_valid
(ensemble: str, valid_models, valid_trainers, valid_plans)
nnunet/inference/pretrained_models/collect_pretrained_models.py:79
↓ 2 callersFunctioncheck_same
checking initial vs corrected dataset :param img1: :param img2: :return:
nnunet/dataset_conversion/Task115_COVIDSegChallenge.py:211
↓ 2 callersMethodcompute_approx_vram_consumption
(patch_size, base_num_features, max_num_features, num_modalities, num_
nnunet/network_architecture/generic_modular_UNet.py:342
↓ 2 callersMethodcompute_loss
(self, output, target)
nnunet/training/network_training/nnUNetTrainerV2_DDP.py:247
↓ 2 callersFunctioncopy_ensembles
(taskname, output_folder, valid_models=('2d', '3d_fullres', '3d_lowres', '3d_cascade_fullres'),
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"
nnunet/inference/pretrained_models/collect_pretrained_models.py:57
↓ 2 callersFunctioncreate_lists_from_splitted_dataset
(base_folder_splitted)
nnunet/experiment_planning/utils.py:82
↓ 2 callersFunctioncreate_region_from_mask
(mask, join_labels: tuple)
nnunet/evaluation/region_based_evaluation.py:34
↓ 2 callersMethodcrop_from_list_of_files
(data_files, seg_file=None)
nnunet/preprocessing/cropping.py:153
↓ 2 callersMethoddetermine_normalization_scheme
(self)
nnunet/experiment_planning/experiment_planner_baseline_3DUNet.py:359
↓ 2 callersMethoddetermine_normalization_scheme
(self)
nnunet/experiment_planning/alternative_experiment_planning/normalization/experiment_planner_3DUNet_CT2.py:35
↓ 2 callersMethoddo_split
(self)
nnunet/training/network_training/nnUNetTrainerCascadeFullRes.py:60
↓ 2 callersMethoddo_split
(self)
nnunet/training/network_training/nnUNetTrainerV2_CascadeFullRes.py:59
↓ 2 callersFunctionevaluate_BraTS_folder
(folder_pred, folder_gt, num_processes: int = 24, strict=False)
nnunet/dataset_conversion/Task082_BraTS_2020.py:385
↓ 2 callersFunctionevaluate_folder
writes a summary.json to folder_with_predictions :param folder_with_gts: folder where the ground truth segmentations are saved. Must be nifti
nnunet/evaluation/evaluator.py:446
↓ 2 callersFunctiongenerate_overlay
image must be a color image, so last dimension must be 3. if image is grayscale, tile it first! Segmentation must be label map of same shape
nnunet/utilities/overlay_plots.py:46
↓ 2 callersFunctionget_case_identifiers
(folder)
nnunet/training/dataloading/dataset_loading.py:26
↓ 2 callersMethodget_classes
(self)
nnunet/experiment_planning/DatasetAnalyzer.py:109
↓ 2 callersFunctionget_identifiers_from_splitted_files
(folder: str)
nnunet/dataset_conversion/utils.py:22
↓ 2 callersFunctionget_insaneDA_augmentation2
(dataloader_train, dataloader_val, patch_size, params=default_3D_augmentation_params,
nnunet/training/data_augmentation/data_augmentation_insaneDA2.py:42
↓ 2 callersFunctionget_mean_foreground_dice
(json_file)
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
nnunet/training/data_augmentation/data_augmentation_noDA.py:31
↓ 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
nnunet/experiment_planning/experiment_planner_baseline_3DUNet.py:144
↓ 2 callersMethodget_properties_for_stage
nnunet/experiment_planning/alternative_experiment_planning/patch_size/experiment_planner_3DUNet_isotropic_in_mm.py:37
↓ 2 callersMethodget_target_spacing
(self)
nnunet/experiment_planning/experiment_planner_baseline_3DUNet.py:66
↓ 2 callersMethodget_target_spacing
per default we use the 50th percentile=median for the target spacing. Higher spacing results in smaller data and thus faster and easi
nnunet/experiment_planning/alternative_experiment_planning/target_spacing/experiment_planner_baseline_3DUNet_targetSpacingForAnisoAxis.py:27
↓ 2 callersFunctionincrease_batch_size
(plans_file: str, save_as: str, bs_factor: int)
nnunet/dataset_conversion/Task115_COVIDSegChallenge.py:27
↓ 2 callersMethodinitialize
:param training: :return:
nnunet/training/network_training/nnUNetTrainerV2_DDP.py:134
↓ 2 callersMethodinitialize_network
inference_apply_nonlin to sigmoid
nnunet/training/network_training/competitions_with_custom_Trainers/BraTS2020/nnUNetTrainerV2BraTSRegions.py:82
↓ 2 callersFunctioninstall_model_from_zip_file
(zip_file: str)
nnunet/inference/pretrained_models/download_pretrained_model.py:300
↓ 2 callersMethodload_properties_of_cropped
(self, case_identifier)
nnunet/experiment_planning/experiment_planner_baseline_3DUNet.py:377
↓ 2 callersMethodload_properties_of_cropped
(self, case_identifier)
nnunet/experiment_planning/DatasetAnalyzer.py:45
↓ 2 callersMethodmaybe_update_lr
(self, epoch=None)
nnunet/training/network_training/nnUNet_variants/optimizer_and_lr/nnUNetTrainerV2_warmup.py:26
↓ 2 callersFunctionplan_and_preprocess
(task_string, processes_lowres=default_num_threads, processes_fullres=3, no_preprocessing=False)
nnunet/experiment_planning/utils.py:144
↓ 2 callersMethodplan_experiment
(self)
nnunet/experiment_planning/experiment_planner_baseline_2DUNet.py:90
↓ 2 callersFunctionprecision
TP / (TP + FP)
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
nnunet/network_architecture/neural_network.py:73
↓ 2 callersMethodpredict_preprocessed_data_return_seg_and_softmax
:param data: :param do_mirroring: :param mirror_axes: :param use_sliding_window: :param step_size: :p
nnunet/training/network_training/nnUNetTrainer.py:483
↓ 2 callersMethodpredict_preprocessed_data_return_seg_and_softmax
(self, data: np.ndarray, do_mirroring: bool = True, m
nnunet/training/network_training/nnUNet_variants/benchmarking/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:
nnunet/training/network_training/nnUNetTrainer.py:417
↓ 2 callersFunctionpretend_to_be_other_trainer
(folder, new_trainer_name, checkpoints=("model_best.model.pkl", "model_final_checkpoint.model.pkl"))
nnunet/inference/change_trainer.py:23
↓ 2 callersFunctionremove_trailing_slash
(filename: str)
nnunet/utilities/file_endings.py:19
↓ 2 callersMethodresample_and_normalize
data and seg must already have been transposed by transpose_forward. properties are the un-transposed values (spacing etc) :p
nnunet/preprocessing/preprocessing.py:225
↓ 2 callersMethodreset
(self)
nnunet/evaluation/metrics.py:51
↓ 2 callersFunctionreverse_axes
(image)
nnunet/dataset_conversion/Task056_Verse_normalize_orientation.py:25
↓ 2 callersMethodrun
(self, target_spacings, input_folder_with_cropped_npz, output_folder, data_identifier, num_threads
nnunet/preprocessing/preprocessing.py:584
↓ 2 callersFunctionrun_in_folder
(folder)
nnunet/evaluation/add_mean_dice_to_json.py:42
↓ 2 callersMethodrun_online_evaluation
(self, tp_hard, fp_hard, fn_hard)
nnunet/training/network_training/nnUNetTrainerV2_DP.py:218
↓ 2 callersMethodrun_online_evaluation
(self, output, target)
nnunet/training/network_training/nnUNet_variants/benchmarking/nnUNetTrainerV2_dummyLoad.py:143
↓ 2 callersMethodrun_online_evaluation
(self, output, target)
nnunet/training/network_training/nnUNet_variants/benchmarking/nnUNetTrainerV2_2epochs.py:219
↓ 2 callersMethodsave_debug_information
(self)
nnunet/training/network_training/nnUNetTrainer.py:297
↓ 2 callersMethodsave_properties_of_cropped
(self, case_identifier, properties)
nnunet/experiment_planning/experiment_planner_baseline_3DUNet.py:373
↓ 2 callersFunctionselect_annotated_frames_mms
(data_folder, out_folder, add_zeros=False, mode='mnms', df_path="/media/full/tera2/data/challenges/mms/Trainin
nnunet/dataset_conversion/Task114_heart_MNMs.py:64
↓ 2 callersFunctionsensitivity
TP / (TP + FN)
nnunet/evaluation/metrics.py:159
↓ 2 callersMethodset_device
(self, device)
nnunet/network_architecture/neural_network.py:38
↓ 2 callersMethodset_labels
Set the labels. :param labels= may be a dictionary (int->str), a set (of ints), a tuple (of ints) or a list (of ints). Labels will onl
nnunet/evaluation/evaluator.py:109
↓ 2 callersMethodsetup_DA_params
(self)
nnunet/training/network_training/nnUNetTrainerCascadeFullRes.py:90
↓ 2 callersMethodsetup_DA_params
- we increase roation angle from [-15, 15] to [-30, 30] - scale range is now (0.7, 1.4), was (0.85, 1.25) - we don't do elast
nnunet/training/network_training/MAMLTrainerV2.py:353
↓ 2 callersMethodsetup_DA_params
(self)
nnunet/training/network_training/nnUNet_variants/data_augmentation/nnUNetTrainerV2_DA3.py:31
↓ 2 callersFunctionspecificity
TN / (TN + FP)
nnunet/evaluation/metrics.py:183
↓ 2 callersFunctionsplit_4d
(input_folder, num_processes=default_num_threads, overwrite_task_output_id=None)
nnunet/experiment_planning/utils.py:31
↓ 2 callersFunctionsplit_4d_for_all_pat
(files_paths, split_folder)
nnunet/dataset_conversion/Task114_heart_MNMs.py:179
↓ 2 callersFunctionswap_flip_dimensions
(cosine_matrix, image, header=None)
nnunet/dataset_conversion/Task056_Verse_normalize_orientation.py:60
↓ 2 callersMethodupdate_fold
used to swap between folds for inference (ensemble of models from cross-validation) DO NOT USE DURING TRAINING AS THIS WILL NOT UPDAT
nnunet/training/network_training/nnUNetTrainer.py:134
← previousnext →101–200 of 895, ranked by callers