happyfunction 2019-06-19 23:50 采纳率: 0%
浏览 952
已结题

怎样用keras实现从预训练模型中提取多层特征?

图片说明

我想从一个预训练的卷积神经网络的不同层中提取特征,然后把这些不同层的特征拼接在一起,实现如上图一样的网络结构,我写的代码如下

    base_model = VGGFace(model='resnet50', include_top=False)
    model1 = base_model
    model2 = base_model

    input1 = Input(shape=(197,197,3))
    model1_out = model1.layers[-12].output
    model1_in = model1.layers[0].output
    model1 = Model(model1_in,model1_out)
    x1 = model1(input1)
    x1 = GlobalMaxPool2D()(x1)


    x2 = model2(input1)
    x2 = GlobalMaxPool2D()(x2)
    out = Concatenate(axis=-1)([x1,x2])
    out = Dense(1,activation='sigmoid')(out)
    model3 = Model([input1,input2],out)

    from keras.utils import plot_model
    plot_model(model3,"model3.png")
    import matplotlib.pyplot as plt
    img = plt.imread('model3.png')
    plt.imshow(img)

但模型可视化显示如下,两个网络的权值并不共享。图片说明

  • 写回答

1条回答

  • threenewbee 2019-06-20 10:41
    关注

    你这个用keras的函数模式就可以实现
    https://www.jianshu.com/p/ee2a170b3a12

    评论

报告相同问题?

悬赏问题

  • ¥15 求差集那个函数有问题,有无佬可以解决
  • ¥15 【提问】基于Invest的水源涵养
  • ¥20 微信网友居然可以通过vx号找到我绑的手机号
  • ¥15 寻一个支付宝扫码远程授权登录的软件助手app
  • ¥15 解riccati方程组
  • ¥15 display:none;样式在嵌套结构中的已设置了display样式的元素上不起作用?
  • ¥15 使用rabbitMQ 消息队列作为url源进行多线程爬取时,总有几个url没有处理的问题。
  • ¥15 Ubuntu在安装序列比对软件STAR时出现报错如何解决
  • ¥50 树莓派安卓APK系统签名
  • ¥65 汇编语言除法溢出问题