Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Mehrdad-Noori/Brain-Tumor-Segmentation
/ functions
Functions
30 in github.com/Mehrdad-Noori/Brain-Tumor-Segmentation
⨍
Functions
30
◇
Types & classes
1
↓ 3 callers
Function
conv_layers
(x, dim)
model.py:144
↓ 3 callers
Function
res_block
(x, dim, encoder_path=True)
model.py:154
↓ 2 callers
Method
flip_axis
(self, x, axis)
data_generator.py:207
↓ 1 callers
Function
apply_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 callers
Method
apply_transform
(self, x, transform_parameters)
data_generator.py:215
↓ 1 callers
Function
create_table
Reads and saves all brain volumes into a single table file. Parameters ---------- dataset_dir : The path to all
prepare_data.py:72
↓ 1 callers
Method
data_load_and_preprocess
Generates data containing batch_size samples
data_generator.py:137
↓ 1 callers
Function
generalized_dice
Generalized Dice Score https://arxiv.org/pdf/1707.03237
loss.py:5
↓ 1 callers
Function
generalized_dice_loss
(y_true, y_pred)
loss.py:23
↓ 1 callers
Method
get_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 callers
Method
get_random_transform
(self)
data_generator.py:232
↓ 1 callers
Function
level_block
(x, dim, level, inc)
model.py:111
↓ 1 callers
Function
level_block_modified
(x, dim, level, inc)
model.py:126
↓ 1 callers
Method
normalize_modalities
Performs normalization on each modalities of input
data_generator.py:194
↓ 1 callers
Function
normalize_slice
Removes 1% of the top and bottom intensities and perform normalization on the input 2D slice.
predict.py:71
↓ 1 callers
Method
normalize_slice
Removes 1% of the top and bottom intensities and perform normalization on the input 2D slice.
data_generator.py:178
↓ 1 callers
Function
normalize_volume
Perform a slice-based normalization on each modalities of input volume.
predict.py:88
↓ 1 callers
Method
on_epoch_end
Updates indexes after each epoch
data_generator.py:123
↓ 1 callers
Function
read_brain
A function that reads and crops a brain modalities (nii.gz format) Parameters ---------- brain_dir : string The path to
predict.py:8
↓ 1 callers
Function
read_brain
A function that reads and crops a brain modalities (nii.gz format) Parameters ---------- brain_dir : string The pa
prepare_data.py:9
↓ 1 callers
Method
read_data
Reads data from table with respect to the 'view'
data_generator.py:164
↓ 1 callers
Function
save_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 callers
Function
se_block
creates a squeeze and excitation block https://arxiv.org/abs/1709.01507 Parameters ---------- x : tensor Input keras
model.py:75
↓ 1 callers
Function
train_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 callers
Function
transform_matrix_offset_center
(matrix, x, y)
data_generator.py:342
↓ 1 callers
Function
unet_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
Function
custom_loss
The final loss function consists of the summation of two losses "GDL" and "CE" with a regularization term.
loss.py:27