Tensorflow Xception广播输入数组错误

问题描述

我正在使用tensorflow-gpu 2.1，并且正在对850x550图像（3通道）进行图像分类。

模型（初步）看起来像这样（使用顺序API）：

input_tensor_def = Input(shape=(850,550,3))
model = Sequential()
xception = Xception(include_top = False,weights = None,input_tensor =  input_tensor_def)

model.add(xception)
model.add(GlobalAvgPool2D())
model.add(Flatten())
model.add(Dense(512,activation='relu'))
model.add(Dense(2,activation='softmax'))

使用模型API，如下所示：

model_core = Xception(weights = None,include_top = False,input_tensor = input_tensor_def)

model_head = model_core.output
model_head = GlobalAvgPool2D()(model_head)
model_head = Flatten()(model_head)
model_head = Dense(512,activation = 'relu')(model_head)
model_head = Dense(2,activation = 'softmax')(model_head)

model = Model(inputs = model_core.input,outputs = model_head

我遇到以下错误：ValueError: Could not broadcast input array from shape (850,3) into shape (850,3,3)

我真的很困惑为什么它试图将高度张量解释为批处理索引。

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）