Ошибка значения: формы (1, 1, 512, 4) и (2, 512, 1, 1) несовместимы, как исправить эту ошибку?

Question

Я не знаю много по-английски sh, так что это может быть написано неуклюже. Мой пользовательский код выдает ошибку при распознавании. Я пытался изменить методы анализа и расположение файлов. Но ничего не помогло. Я не супер программист, поэтому я не совсем понимаю, что происходит в глубине кода. Я также не очень понимаю, где выходное изображение должно быть сохранено и как оно будет называться. Код программы:

from imageai.Prediction.Custom import CustomImagePrediction
import os

execution_path = os.getcwd()

prediction = CustomImagePrediction()
prediction.setModelTypeAsSqueezeNet()
prediction.setModelPath(os.path.join(execution_path, "model_ex-065_acc-0.875000.h5"))
prediction.setJsonPath(os.path.join(execution_path, "model_class.json"))
prediction.loadModel(num_objects=4)

predictions, probabilities = prediction.predictImage(os.path.join(execution_path, "ScreenHack.jpg"), result_count=5)

for eachPrediction, eachProbability in zip(predictions, probabilities):
    print(eachPrediction , " : " , eachProbability)

Текст ошибки:

ValueError: Shapes (1, 1, 512, 4) and (2, 512, 1, 1) are incompatible

Answer 1

Фактическое количество элементов, которое вы можете получить, умножив все значения, для первого он будет ставить 2048, для второго - только 1024. Поэтому его нельзя изменить.