去除keras模型第一层出错答案

【问题标题】：Error in removing the first layer of keras model去除keras模型第一层出错
【发布时间】：2020-03-11 13:28:56
【问题描述】：

import numpy as np
from keras.applications.vgg19 import decode_predictions
from prettytable import PrettyTable
import time
from keras import backend as K
from tensorflow import keras
from tensorflow.python import keras
from keras import models, layers, Model, Input
import tensorflow as tf

model_2=keras.models.load_model('model_2.h5',compile=False)
model_2.summary()

predictions1= np.load('D:/predictions_result.npy')
def profiler(model, test_input):
 data_input=test_input
 for layer in model.layers:
    start = time.time()
    im_imput=keras.layers.Input(batch_shape=model.get_layer(layer.name).get_input_shape_at(0))
    im_out = layer(im_imput)
    new_model = keras.models.Model(inputs=im_imput,outputs=im_out)
    data_input = new_model.predict(data_input)
    end = time.time() - start
    print(end)
    result=1


profiler(model_2,predictions1)

tmp=np.zeros((1,224,224,64))

for i in range(0,1):
   tmp[i,:,:,:]=predictions1[i,:]

predictions2 = model_2.predict(tmp)
label_vgg19 = decode_predictions(predictions2)
print ('label_vgg19 =', label_vgg19)

当我尝试运行上述代码时，出现以下错误。我的问题是如何在加载后删除模型的第一层。我最初将 VGG 模型拆分为子模型，然后加载子模型。我尝试了不同的方法，但都没有奏效。非常感谢您的帮助。

  Traceback (most recent call last):
  File "C:/Users/40227422/PycharmProjects/model_partititon/model_2_sock.py", line 42, in <module>
  profiler(model_2,predictions1)
  File "C:/Users/40227422/PycharmProjects/model_partititon/model_2_sock.py", line 28, in the 
    profiler
  data_input = new_model.predict(data_input)
  File "C:\Users\40227422\AppData\Local\Continuum\miniconda3\envs\tensorflow\lib\site- 
  packages\tensorflow\python\keras\engine\training_utils.py", line 332, in standardize_input_data
  ' but got array with shape ' + str(data_shape))
  ValueError: Error when checking input: expected input_1 to have shape (224, 224, 3) but got array 
  with shape (224, 224, 64)

当我尝试使用 kerassurgeon 使用以下代码删除图层时出现错误 ValueError: 没有足够的值来解包（预期 2，得到 0）

 from kerassurgeon import Surgeon
 surgeon = Surgeon(model_2)
 layer_1 = model_2.layers[0] # selecting 2nd layer

 surgeon.add_job('delete_layer', layer_1)
 new_model = surgeon.operate()

【问题讨论】：

我现在也遇到了类似的问题，你知道是哪里出了问题吗？

标签： tensorflow conv-neural-network keras-layer tf.keras keras-2

【解决方案1】：

我无法重现您所面临的错误。可能是您可以共享可重现的代码。下面的选项我尝试删除一个图层并且它有效。

from kerassurgeon.operations import delete_layer
# delete layer_1 from a model
model = delete_layer(model_2, layer_1)

或

# delete layer_1 from a model
from kerassurgeon import Surgeon
surgeon = Surgeon(model_2)
surgeon.add_job('delete_layer', layer_1)
new_model = surgeon.operate()

【讨论】：