ValueError：形状 (1, 1, 512, 4) 和 (2, 512, 1, 1) 不兼容如何解决此错误？

Question

我不太会英文，所以可能写得很笨拙。 我的自定义代码在识别时出错。我试图改变分析方法和文件位置。但没有任何帮助。我不是超级程序员，所以我不太了解代码深处发生了什么。我也不太明白输出图像应该保存在哪里以及它将被称为什么。 程序代码：

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。这就是它无法重塑的原因。