newnet.py 文件源码

python
阅读 27 收藏 0 点赞 0 评论 0

项目:head-segmentation 作者: szywind 项目源码 文件源码
def vgg16():
    base_model = VGG16(weights=None, include_top=False, input_shape = (224,224,3))
    # Classification block
    x = Flatten(name='flatten', input_shape=base_model.output_shape[1:])(base_model.output)
    x = Dense(512, activation='relu', name='fc1')(x)
    x = Dense(512, activation='relu', name='fc2')(x)
    x = Dense(17, activation='softmax', name='predictions')(x)

    model = Model(inputs=base_model.input, outputs=x)
    return model
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号