MCPcopy Create free account

hub / github.com/SimJeg/FC-DenseNet / functions

Functions17 in github.com/SimJeg/FC-DenseNet

↓ 4 callersFunctionBN_ReLU_Conv
Apply successivly BatchNormalization, ReLu nonlinearity, Convolution and Dropout (if dropout_p > 0) on the inputs
layers.py:8
↓ 4 callersFunctionbatch_loop
Loop on the batches
train.py:21
↓ 3 callersMethodrestore
Load the weights
FC-DenseNet.py:122
↓ 3 callersMethodsummary
Print a summary of the network architecture
FC-DenseNet.py:129
↓ 2 callersFunctionload_data
(dataset, train_crop_size=(224, 224), one_hot=False, batch_size=10, horizontal_fli
data_loader.py:4
↓ 2 callersFunctiontheano_metrics
Returns the intersection I and union U (to compute the jaccard I/U) and the accuracy. :param y_pred: tensor of predictions. shape (b*0*1, c
metrics.py:4
↓ 1 callersFunctionSoftmaxLayer
Performs 1x1 convolution followed by softmax nonlinearity The output will have the shape (batch_size * n_rows * n_cols, n_classes)
layers.py:48
↓ 1 callersFunctionTransitionDown
Apply first a BN_ReLu_conv layer with filter size = 1, and a max pooling with a factor 2
layers.py:21
↓ 1 callersFunctionTransitionUp
Performs upsampling on block_to_upsample by a factor 2 and concatenates it with the skip_connection
layers.py:32
↓ 1 callersFunctioncrossentropy
(y_pred, y_true, void_labels)
metrics.py:66
↓ 1 callersFunctioninitiate_training
(cf)
train.py:198
↓ 1 callersMethodsave
Save the weights
FC-DenseNet.py:118
↓ 1 callersFunctiontest
This function builds the model defined in config_path and restores the weights defined in weight_path. It then reports the jaccard and global
test.py:12
↓ 1 callersFunctiontrain
(cf)
train.py:55
Method__init__
This code implements the Fully Convolutional DenseNet described in https://arxiv.org/abs/1611.09326 The network consist of a downsamp
FC-DenseNet.py:12
Methodfilter_function
We only display the layers in the list below
FC-DenseNet.py:134
Functionnumpy_metrics
Similar to theano_metrics to metrics but instead y_pred and y_true are now numpy arrays
metrics.py:40