kerasext.py 文件源码

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

项目:kaggle_amazon 作者: asanakoy 项目源码 文件源码
def create_resnet50(input_img):
    net = ResNet50(weights='imagenet', include_top=False,
                      input_tensor=input_img)

    for layer in net.layers[1:]:
        layer.trainable = False
    net = Reshape((-1,))(net.outputs[0])
    return net
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号