Я получаю эту ошибку на этапе обучения cnn, что-то не так с измерениями набора данных, которые я предполагаю. Я импортировал скрипт python vggnet в свой обучающий скрипт, и с тех пор я получаю сообщение об ошибке, но как избавиться отэто? как я новичок в 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))
мой набор данных выглядит так: 