MCPcopy Create free account

hub / github.com/Mehrdad-Noori/Brain-Tumor-Segmentation / functions

Functions30 in github.com/Mehrdad-Noori/Brain-Tumor-Segmentation

↓ 3 callersFunctionconv_layers
(x, dim)
model.py:144
↓ 3 callersFunctionres_block
(x, dim, encoder_path=True)
model.py:154
↓ 2 callersMethodflip_axis
(self, x, axis)
data_generator.py:207
↓ 1 callersFunctionapply_affine_transform
Applies an affine transformation specified by the parameters given. # Arguments x: 2D numpy array, single image. theta: Rotat
data_generator.py:260
↓ 1 callersMethodapply_transform
(self, x, transform_parameters)
data_generator.py:215
↓ 1 callersFunctioncreate_table
Reads and saves all brain volumes into a single table file. Parameters ---------- dataset_dir : The path to all
prepare_data.py:72
↓ 1 callersMethoddata_load_and_preprocess
Generates data containing batch_size samples
data_generator.py:137
↓ 1 callersFunctiongeneralized_dice
Generalized Dice Score https://arxiv.org/pdf/1707.03237
loss.py:5
↓ 1 callersFunctiongeneralized_dice_loss
(y_true, y_pred)
loss.py:23
↓ 1 callersMethodget_brain_idx
Getting the brain indexes that will be used by the generator. if mode=='train' => the original indexes will be used (because we bui
data_generator.py:87
↓ 1 callersMethodget_random_transform
(self)
data_generator.py:232
↓ 1 callersFunctionlevel_block
(x, dim, level, inc)
model.py:111
↓ 1 callersFunctionlevel_block_modified
(x, dim, level, inc)
model.py:126
↓ 1 callersMethodnormalize_modalities
Performs normalization on each modalities of input
data_generator.py:194
↓ 1 callersFunctionnormalize_slice
Removes 1% of the top and bottom intensities and perform normalization on the input 2D slice.
predict.py:71
↓ 1 callersMethodnormalize_slice
Removes 1% of the top and bottom intensities and perform normalization on the input 2D slice.
data_generator.py:178
↓ 1 callersFunctionnormalize_volume
Perform a slice-based normalization on each modalities of input volume.
predict.py:88
↓ 1 callersMethodon_epoch_end
Updates indexes after each epoch
data_generator.py:123
↓ 1 callersFunctionread_brain
A function that reads and crops a brain modalities (nii.gz format) Parameters ---------- brain_dir : string The path to
predict.py:8
↓ 1 callersFunctionread_brain
A function that reads and crops a brain modalities (nii.gz format) Parameters ---------- brain_dir : string The pa
prepare_data.py:9
↓ 1 callersMethodread_data
Reads data from table with respect to the 'view'
data_generator.py:164
↓ 1 callersFunctionsave_predicted_results
Save the segmented results into a .nii.gz file, so that it can be uploaded to the BraTS server. Note that to correctly save the segmented bra
predict.py:102
↓ 1 callersFunctionse_block
creates a squeeze and excitation block https://arxiv.org/abs/1709.01507 Parameters ---------- x : tensor Input keras
model.py:75
↓ 1 callersFunctiontrain_model
The function that builds/loads UNet model, initializes the data generators for training and validation, and finally trains the model.
train.py:9
↓ 1 callersFunctiontransform_matrix_offset_center
(matrix, x, y)
data_generator.py:342
↓ 1 callersFunctionunet_model
Builds UNet model Parameters ---------- input_shape : tuple Shape of the input data (height, width, channel) modifie
model.py:12
Method__getitem__
(self, index)
data_generator.py:114
Method__init__
Custom data generator based on Keras Sequance class. This implementation enables multiprocessing and on-the-fly data augmentation
data_generator.py:13
Method__len__
(self)
data_generator.py:110
Functioncustom_loss
The final loss function consists of the summation of two losses "GDL" and "CE" with a regularization term.
loss.py:27