model_fcn.py 文件源码

def testnet_fcn(n_classes):
    stride = 32
    input_tensor = Input(shape=(None, None, 3))
    x = Convolution2D(4,5,5,name='conv',
                    activation = 'relu', border_mode='same', subsample= (stride,stride))(input_tensor)

    x = Softmax4D(axis=-1)(x)
    x = UpSampling2D(size=(stride,stride))(x)
    x = Convolution2D(n_classes,3,3,name = 'pred_up',border_mode = 'same')(x)

    model = Model(input=input_tensor,output=x)

    return model, stride