如何解决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 举报,一经查实,本站将立刻删除。