ValueError:应该在输入列表上调用合并层。Tensorflow Keras
ValueError:应该在输入列表上调用合并层。Tensorflow Keras
提问于 2019-06-26 19:14:33
我目前正在尝试使用MobileNetV2的前50层。因此，我想提取这些层并创建一个新模型。
我以为我可以调用每一层，但是"block_2_add“层会导致错误，我不明白为什么。
import tensorflow as tf
from keras.models import Model
mobile_net=tf.keras.applications.mobilenet_v2.MobileNetV2(input_shape=(224,224,3), alpha=0.5, include_top=False, weights='imagenet')
inputs = Input(shape=(224, 224, 3))
x=mobile_net.layers[1](inputs)
for layer in mobile_net.layers[2:50]:
x=layer(x)
{'name': 'block_2_add', 'trainable': True, 'dtype': 'float32'}
---------------------------------------------------------------------------
ValueError Traceback (most recent call last)
<ipython-input-77-5873b9344fa3> in <module>()
3 for layer in mobile_net.layers[2:50]:
4 print(layer.get_config())
----> 5 x=layer(x)
6
7 for layer in mobile_net.layers[:50]:
1 frames
/usr/local/lib/python3.6/dist-packages/tensorflow/python/keras/layers/merge.py in call(self, inputs)
119 def call(self, inputs):
120 if not isinstance(inputs, list):
--> 121 raise ValueError('A merge layer should be called on a list of inputs.')
122 if self._reshape_required:
123 reshaped_inputs = []
ValueError: A merge layer should be called on a list of inputs.回答 2
Stack Overflow用户
发布于 2019-06-26 19:37:22
我的猜测是，MobileNetV2不是顺序模型，即层图不是线性的。如果你只想要模型的输出，而不是任何中间层的输出，我认为下面的代码应该可以完成这项工作(即使你似乎想在输出之前计算最后一层，结果仍然应该是你想要的)：
import tensorflow as tf
from keras.models import Model
mobile_net=tf.keras.applications.mobilenet_v2.MobileNetV2(input_shape=(224,224,3), alpha=0.5, include_top=False, weights='imagenet')
inputs = Input(shape=(224, 224, 3))
output = mobile_net(inputs)Stack Overflow用户
发布于 2021-05-30 10:04:55
我可能会迟到，但我猜下面的代码将为您做这件事
pre_trained_model = MobileNetV2(input_shape = (256, 256, 3),
include_top = False,
weights = "imagenet" )
last_layer = pre_trained_model.get_layer('block_15_project_BN'#the name of the last layer you want from the model)
last_output = last_layer.output
input_l = pre_trained_model.input
base_model1 = tf.keras.Model(input_l, last_output
)页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/56771548
复制相关文章
相似问题
腾讯云开发者
