Question

我建立了一个交通信号灯检测模型，其他所有东西都工作正常。但是当我尝试以可视方式显示预测和基本事实时，它给了我一个类似于上面的错误。我猜想预测函数是错误的。有人可以帮忙吗？下面是代码。

# Display the predictions and the ground truth visually
   fig = plt.figure(figsize=(10, 10))
   j = 1
   for i in range(0, 1000, 100):
       truth = test_labels_a[i]
       prediction = model.predict(i)
       plt.subplot(5, 2, j)
       j = j + 1
       plt.axis('off')
       color = 'green' if truth == prediction else 'red'
       plt.text(40, 10, "Truth:        {0}\nPrediction: {1}".format(truth, prediction),
                fontsize=12, color=color)
       plt.imshow(test_x[i], cmap='gray')

Answer 1

model.predict(i)应该类似于model.predict(test_data)

AttributeError：“ int”对象没有属性“ ndim”

1 个答案: