使用VGGNET脚本训练CNN时出现值错误

Question

我在cnn的训练阶段收到此错误，我假设的数据集维度有问题。我已将vggnet python脚本导入到我的训练脚本中，此后我立即得到了错误，但如何摆脱此错误因为我是ML的新手，并且是第一次使用CNN。

我已经研究了先前为同一问题提供的解决方案，但似乎没有得到答案。

# initialize our VGG-like Convolutional Neural Network
model = Smallvggnet.build(width=200, height=200, depth=3,
classes=len(lb.classes_))

print("[INFO] training network...")
opt = SGD(lr=INIT_LR, decay=INIT_LR / EPOCHS)
model.compile(loss="categorical_crossentropy", optimizer=opt,
metrics=["accuracy"])

 # train the network
 H = model.fit_generator(aug.flow(trainX, trainY, batch_size=BS),
validation_data=(testX, testY), steps_per_epoch=len(trainX) // BS,
epochs=EPOCHS)

我希望对网络进行培训，但这会给我带来Value错误： ValueError ：（“ NumpyArrayIterator中的输入数据应该排在第4位。您传递了一个具有形状的数组”，（244，200，200））

我的数据集如下所示：