MCPcopy Create free account

hub / github.com/JunMa11/SegLossOdyssey / functions

Functions250 in github.com/JunMa11/SegLossOdyssey

↓ 37 callersMethodprint_to_log_file
(self, *args, also_print_to_console=True, add_timestamp=True)
test/nnUNetV1/network_training/network_trainer.py:196
↓ 30 callersFunctionsum_tensor
(inp, axes, keepdim=False)
losses_pytorch/dice_loss.py:20
↓ 13 callersMethod__init__
Sensitivity-Specifity loss paper: http://www.rogertam.ca/Brosch_MICCAI_2015.pdf tf code: https://github.com/NifTK/NiftyNet/bl
test/nnUNetV1/loss_functions/dice_loss.py:187
↓ 11 callersMethod__init__
square_volumes will square the weight term. The paper recommends square_volumes=True; I don't (just an intuition)
test/nnUNetV2/loss_functions/dice_loss.py:27
↓ 11 callersMethod__init__
Sensitivity-Specifity loss paper: http://www.rogertam.ca/Brosch_MICCAI_2015.pdf tf code: https://github.com/NifTK/NiftyNet
losses_pytorch/dice_loss.py:193
↓ 4 callersMethod__init__
compute boudary loss only compute the loss of foreground ref: https://github.com/LIVIAETS/surface-loss/blob/108bd9892adca476e
test/nnUNetV1/loss_functions/boundary_loss.py:132
↓ 4 callersFunctionget_tp_fp_fn
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,
test/nnUNetV1/loss_functions/dice_loss.py:25
↓ 4 callersFunctionget_tp_fp_fn
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
losses_pytorch/dice_loss.py:31
↓ 4 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,
test/nnUNetV2/loss_functions/dice_loss.py:101
↓ 4 callersMethodinitialize
create self.output_folder modify self.output_folder if you are doing cross-validation (one folder per fold) set self.tr_gen
test/nnUNetV1/network_training/network_trainer.py:102
↓ 4 callersMethodload_checkpoint
(self, fname, train=True)
test/nnUNetV1/network_training/network_trainer.py:272
↓ 4 callersMethodload_dataset
(self)
test/nnUNetV1/network_training/nnUNetTrainer.py:314
↓ 4 callersMethodrun_iteration
(self, data_generator, do_backprop=True, run_online_evaluation=False)
test/nnUNetV1/network_training/network_trainer.py:505
↓ 4 callersMethodsave_checkpoint
(self, fname, save_optimizer=True)
test/nnUNetV1/network_training/network_trainer.py:229
↓ 4 callersFunctionsoftmax_helper
(x)
losses_pytorch/dice_loss.py:12
↓ 3 callersMethod__init__
compute boudary loss only compute the loss of foreground ref: https://github.com/LIVIAETS/surface-loss/blob/108bd9892adca4
losses_pytorch/boundary_loss.py:83
↓ 3 callersMethodinitialize_network_optimizer_and_scheduler
This is specific to the U-Net and must be adapted for other network architectures :return:
test/nnUNetV1/network_training/nnUNetTrainer.py:207
↓ 3 callersMethodload_plans_file
This is what actually configures the entire experiment. The plans file is generated by experiment planning :return:
test/nnUNetV1/network_training/nnUNetTrainer.py:262
↓ 3 callersMethodpredict_preprocessed_data_return_softmax
Don't use this. If you need softmax output, use preprocess_predict_nifti and set softmax_output_file. :param data: :param do_
test/nnUNetV1/network_training/nnUNetTrainer.py:379
↓ 3 callersFunctionsum_tensor
(inp, axes, keepdim=False)
losses_pytorch/boundary_loss.py:17
↓ 2 callersMethod_maybe_init_amp
(self)
test/nnUNetV1/network_training/network_trainer.py:309
↓ 2 callersMethoddistance_field
(self, img: np.ndarray)
losses_pytorch/hausdorff.py:27
↓ 2 callersMethoddo_split
(self)
test/nnUNetV1/network_training/nnUNetTrainerCascadeFullRes.py:43
↓ 2 callersMethoddo_split
This is a suggestion for if your dataset is a dictionary (my personal standard) :return:
test/nnUNetV1/network_training/network_trainer.py:121
↓ 2 callersFunctionflatten
Flattens a given tensor such that the channel axis is first. The shapes are transformed as follows: (N, C, D, H, W) -> (C, N * D * H * W)
test/nnUNetV1/loss_functions/dice_loss.py:124
↓ 2 callersFunctionflatten
Flattens a given tensor such that the channel axis is first. The shapes are transformed as follows: (N, C, D, H, W) -> (C, N * D * H * W)
losses_pytorch/dice_loss.py:130
↓ 2 callersMethodforward
(self, inp, target)
test/nnUNetV1/loss_functions/ND_Crossentropy.py:81
↓ 2 callersMethodget_basic_generators
(self)
test/nnUNetV1/network_training/nnUNetTrainer.py:317
↓ 2 callersMethodperform_erosion
( self, pred: np.ndarray, target: np.ndarray, debug )
losses_pytorch/hausdorff.py:100
↓ 2 callersMethodsetup_DA_params
(self)
test/nnUNetV1/network_training/nnUNetTrainerCascadeFullRes.py:73
↓ 2 callersMethodsetup_DA_params
(self)
test/nnUNetV1/network_training/nnUNetTrainer.py:128
↓ 2 callersFunctionsoftmax_helper
(x)
losses_pytorch/boundary_loss.py:9
↓ 1 callersMethod__init__
(self, weight=None, ignore_index=-100, threshold=None)
test/nnUNetV1/loss_functions/ND_Crossentropy.py:77
↓ 1 callersMethod__init__
(self, alpha=2.0, **kwargs)
losses_pytorch/hausdorff.py:22
↓ 1 callersMethod__init__
(self, weight=None, ignore_index=-100, k=10)
losses_pytorch/ND_Crossentropy.py:38
↓ 1 callersFunctioncompute_edts_forPenalizedLoss
GT.shape = (batch_size, x,y,z) only for binary segmentation
test/nnUNetV1/loss_functions/ND_Crossentropy.py:207
↓ 1 callersFunctioncompute_edts_forPenalizedLoss
GT.shape = (batch_size, x,y,z) only for binary segmentation
losses_pytorch/ND_Crossentropy.py:151
↓ 1 callersFunctioncompute_edts_forPenalizedLoss
GT.shape = (batch_size, x,y,z) only for binary segmentation
losses_pytorch/boundary_loss.py:175
↓ 1 callersFunctioncompute_gt_dtm
compute the distance transform map of foreground in ground gruth. input: segmentation, shape = (batch_size, class, x, y, z) output: the f
test/nnUNetV1/loss_functions/boundary_loss.py:196
↓ 1 callersFunctioncompute_pred_dtm
compute the distance transform map of foreground in prediction. input: segmentation, shape = (batch_size, class, x, y, z) output: the for
test/nnUNetV1/loss_functions/boundary_loss.py:216
↓ 1 callersFunctioncompute_sdf
compute the signed distance map of binary mask input: segmentation, shape = (batch_size, x, y, z) output: the Signed Distance Map (SDM)
test/nnUNetV1/loss_functions/boundary_loss.py:102
↓ 1 callersMethodfinish_online_evaluation
Can be implemented, does not have to :return:
test/nnUNetV1/network_training/network_trainer.py:545
↓ 1 callersMethodforward
(self, inp, target)
losses_pytorch/ND_Crossentropy.py:42
↓ 1 callersMethodget_basic_generators
(self)
test/nnUNetV1/network_training/nnUNetTrainerCascadeFullRes.py:57
↓ 1 callersFunctionget_default_augmentation_withEDT
(dataloader_train, dataloader_val, patch_size, idx_of_edts, params=defaul
test/nnUNetV1/network_training/nnUNetTrainerEDT.py:23
↓ 1 callersFunctionget_tp_fp_fn
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,
test/nnUNetV1/loss_functions/boundary_loss.py:11
↓ 1 callersFunctionget_tp_fp_fn
copy from: https://github.com/MIC-DKFZ/nnUNet/blob/master/nnunet/training/loss_functions/dice_loss.py net_output must be (b, c, x, y(, z)))
losses_pytorch/boundary_loss.py:28
↓ 1 callersFunctiongt2onehot
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,
test/nnUNetV1/loss_functions/dice_loss.py:287
↓ 1 callersMethodload_best_checkpoint
(self, train=True)
test/nnUNetV1/network_training/network_trainer.py:255
↓ 1 callersMethodload_checkpoint_ram
used for if the checkpoint is already in ram :param saved_model: :param train: :return:
test/nnUNetV1/network_training/network_trainer.py:279
↓ 1 callersFunctionlovasz_grad
Computes gradient of the Lovasz extension w.r.t sorted errors See Alg. 1 in paper
test/nnUNetV1/loss_functions/lovasz_loss.py:7
↓ 1 callersFunctionlovasz_grad
Computes gradient of the Lovasz extension w.r.t sorted errors See Alg. 1 in paper
losses_pytorch/lovasz_loss.py:7
↓ 1 callersMethodlovasz_softmax_flat
(self, inputs, targets)
test/nnUNetV1/loss_functions/lovasz_loss.py:39
↓ 1 callersMethodlovasz_softmax_flat
(self, inputs, targets)
losses_pytorch/lovasz_loss.py:39
↓ 1 callersMethodmanage_patience
(self)
test/nnUNetV1/network_training/network_trainer.py:433
↓ 1 callersMethodmaybe_save_checkpoint
Saves a checkpoint every save_ever epochs. :return:
test/nnUNetV1/network_training/network_trainer.py:396
↓ 1 callersMethodmaybe_update_lr
(self)
test/nnUNetV1/network_training/network_trainer.py:384
↓ 1 callersMethodon_epoch_end
(self)
test/nnUNetV1/network_training/network_trainer.py:483
↓ 1 callersMethodplot_progress
Should probably by improved :return:
test/nnUNetV1/network_training/network_trainer.py:159
↓ 1 callersMethodprepare_kernels
(self)
losses_pytorch/hausdorff.py:92
↓ 1 callersMethodpreprocess_patient
Used to predict new unseen data. Not used for the preprocessing of the training/test data :param input_files: :return:
test/nnUNetV1/network_training/nnUNetTrainer.py:339
↓ 1 callersMethodprob_flatten
(self, input, target)
test/nnUNetV1/loss_functions/lovasz_loss.py:27
↓ 1 callersMethodprob_flatten
(self, input, target)
losses_pytorch/lovasz_loss.py:27
↓ 1 callersMethodprocess_plans
(self, plans)
test/nnUNetV1/network_training/nnUNetTrainerCascadeFullRes.py:69
↓ 1 callersMethodprocess_plans
(self, plans)
test/nnUNetV1/network_training/nnUNetTrainerEDT.py:134
↓ 1 callersMethodprocess_plans
(self, plans)
test/nnUNetV1/network_training/nnUNetTrainer.py:269
↓ 1 callersMethodrun_online_evaluation
Can be implemented, does not have to :param output_torch: :param target_npy: :return:
test/nnUNetV1/network_training/network_trainer.py:536
↓ 1 callersMethodrun_online_evaluation
(self, output, target)
test/nnUNetV1/network_training/nnUNetTrainer.py:534
↓ 1 callersMethodupdate_eval_criterion_MA
If self.all_val_eval_metrics is unused (len=0) then we fall back to using -self.all_val_losses for the MA to determine early stopping
test/nnUNetV1/network_training/network_trainer.py:408
↓ 1 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
test/nnUNetV1/network_training/nnUNetTrainer.py:109
↓ 1 callersMethodupdate_train_loss_MA
(self)
test/nnUNetV1/network_training/network_trainer.py:498
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainerWCET4.py:7
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainer_Focal.py:7
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainerCascadeFullRes.py:18
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainer_ExpLogT1.py:7
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainer_HDBinary.py:6
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainer_DiceHD.py:6
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainerCE.py:6
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainer_Dice.py:7
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainer_Asym.py:7
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainer_SmDice.py:7
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainerWCET0.py:7
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainer_ExpLogT4.py:7
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainerWCET2.py:7
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainer_IouTopK10.py:6
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainerWCET1.py:7
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainer_IouCE.py:7
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainer_GDice.py:8
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainer_FocalTversky.py:7
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainer_DiceBD.py:6
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainer_pGDice.py:7
Method__init__
A generic class that can train almost any neural network (RNNs excluded). It provides basic functionality such as the training loop,
test/nnUNetV1/network_training/network_trainer.py:26
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainerEDT.py:126
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainer_Tversky.py:7
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainer_SS.py:7
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainer_DiceTopK10.py:7
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainer_ExpLogT2.py:7
Method__init__
:param deterministic: :param fold: can be either [0 ... 5) for cross-validation, 'all' to train on all available training data or
test/nnUNetV1/network_training/nnUNetTrainer.py:27
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainer_Lovasz.py:7
Method__init__
(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None,
test/nnUNetV1/network_training/nnUNetTrainer_ExpLogT0.py:7
next →1–100 of 250, ranked by callers