tensorflow - ValueError: 层序号_3 的输入 0 与层不兼容:预期 ndim=5,发现 ndim=4。收到的完整形状:(无、288、432、3)
问题描述
- 从数据集加载的图像
training_set = image_dataset_from_directory(
data_dir,
validation_split=0.3,
subset="training",
seed=123,
image_size=(img_height, img_width),
batch_size=batch_size,
label_mode='categorical',
color_mode='rgb',
shuffle=True,
)
- 模型
num_classes = 10
model = Sequential()
model.add(ConvLSTM2D(filters=64, kernel_size=(3,3), activation='relu',
input_shape=(10,img_height,img_width,3)))
model.add(Dropout(0.2))
model.add(Flatten())
model.add(Dense(100, activation='relu'))
model.add(Dense(num_classes, activation='softmax'))
model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
model.fit_generator()
model.fit_generator(生成器=training_set,validation_data=testing_set,verbose=1,callbacks=callbacks,epochs=120)
解决方案
推荐阅读
- javascript - 模板文字中 JSON 格式的动态 API 中的 For 循环数组
- c - WNOHANG 是否清理子资源
- android - Android:只有数字的单选按钮标签在宽度上分布不均匀
- android - 如何将事件从一个片段传递到另一个片段?
- html - 想要将 html 页面重定向为其他页面的子页面
- slack - 如何通知不在频道中的 Slack 用户?
- django - Django:我不知道我的代码有问题
- python - 如何使用 Tkinter 使框架居中?
- c# - HttpClient 不支持 C# 中的 PostAsJsonAsync 方法和 dll
- python - 数据框中的节点和边列用于构建网络