python - keras VGG19 add Input layer got graph disconnected error
问题描述
我有以下代码,我不明白为什么这是图形断开错误?我不知道出了什么问题。
def create_model_tl_attn_posInputOnly(input_shape):
print("start creating model - transfer learning ...")
emb = embed_encoding2d(input_shape[0], input_shape[1], input_shape[2]) # this is from another class
img_input = Input(shape=input_shape)
base_model = vgg19.VGG19(include_top=False, input_shape=input_shape, weights="imagenet")
base_model.trainable = False
x = base_model(img_input + emb)
flat1 = Flatten()(x)
class1 = Dense(1024, activation='relu')(flat1)
dropout1 = Dropout(0.2)(class1)
class2 = Dense(512, activation='relu')(dropout1)
dropout2 = Dropout(0.2)(class2)
output = Dense(num_classes, activation='softmax')(dropout2)
model = Model(inputs=img_input, outputs=output)
return model
我更改为model = Model(inputs=base_model.inputs, outputs=output)
,但仍然出现graph disconnected
错误。
解决方案
我更新代码如下,没有错误,但是我不知道是否input
是和的总和img_input
,emb
如何在训练期间检查?
def create_model_tl_attn_posInputOnly(input_shape):
print("start creating model - transfer learning ...")
emb = embed_encoding2d(input_shape[0], input_shape[1], input_shape[2]) # this is from another class
img_input = Input(shape=input_shape)
base_model = vgg19.VGG19(include_top=False, input_shape=input_shape, weights="imagenet")
base_model.trainable = False
x = base_model(img_input + emb)
x = base_model.layers[-1].output
flat1 = Flatten()(x)
class1 = Dense(1024, activation='relu')(flat1)
dropout1 = Dropout(0.2)(class1)
class2 = Dense(512, activation='relu')(dropout1)
dropout2 = Dropout(0.2)(class2)
output = Dense(num_classes, activation='softmax')(dropout2)
model = Model(inputs=base_model.inputs, outputs=output)
return model
推荐阅读
- sql - 需要找到两个日期之间的差异以及时间戳,但我需要在 postgresql 中排除星期六和星期日
- jenkins - 詹金斯声明性管道:如何在名称中有空格时访问参数的值?
- c# - 计算指数延迟的估计时间
- apache-kafka - 即使 linger.ms 为 0,Kafka 超时异常
- mysql - 在where子句中使用时在mysql中成功添加序列号但未知列
- amazon-web-services - AWS,我应该将内存限制设置多高?
- python - Bootstrap Modal 不适用于 jinja 2 中的“for”
- reactjs - OnClick 反应按钮以显示表格
- javascript - 未捕获的类型错误:无法读取未定义 React 的属性“过滤器”
- angular - 如何在Angular中注入具有构造函数参数的对象?