DenseNet.py 文件源码

def denseblock(x, nb_layers, nb_filter, growth_rate):
    for i in range(nb_layers):
        if i<=2:
            kernel_size=3
        elif i==3 or i==5:
            kernel_size=1
        else:
            kernel_size=5
        merge_tensor = conv_factory(x, growth_rate,kernel_size)
        #x = merge([merge_tensor, x], mode='concat', concat_axis=-1)
        x=concatenate([merge_tensor, x],axis=-1)
    return x