Keras - model.predict（图片）[0]

Question

我遇到了下面的陈述，并且不确定model.predict(image)[0] 的含义：

[0]

所以，我在没有 [0]和<{1}}一次的情况下打印了上述声明，如下所示：

print model.predict(image)

[[ 0.99335432  0.00664574]]

print model.predict(image)[0]

[ 0.99335432  0.00664574]

是的，我可以看到在后一种情况下已删除外方括号，但仍不确定它的含义。有什么想法吗？

感谢。

Answer 1

model.predict返回Numpy数组。因为API假设有多个输入，如果你提供一个包含N个图像的矩阵，比如dim =（N，H，W，3），model.predict将返回[[ouptut0]，[output1]，... [输出（N-1）]]。

在您的情况下，输入中只有一个图像，因此输入维度实际上是（1，H，W，3），因此model.predict返回[[ 0.99335432 0.00664574]]。

[0]返回model.predict返回的第一个元素，因此[ 0.99335432 0.00664574]。