IntensityMatchMultiResEndoConv.py 文件源码

def create_cnn_network_small(input_dim, no_conv_filt, dense_n):
    '''Base network to be shared (eq. to feature extraction).
    '''
    seq = Sequential()
    kern_size = 3

    # conv layer
    seq.add(Convolution3D(no_conv_filt, kern_size, kern_size, kern_size, input_shape=input_dim,
                          border_mode='valid', dim_ordering='th', activation='relu'))
    seq.add(Dropout(.2))
    seq.add(BatchNormalization(mode=2))

    # dense layer
    seq.add(Flatten())
    seq.add(Dense(dense_n, activation='relu'))
    seq.add(Dropout(.2))
    seq.add(BatchNormalization(mode=2))

    return seq


# train model given x_train and y_train