는 Keras는 - model.predict (이미지) [0]

Question

내가 아래 문을 가로 질러 와서 무엇을 [0]의 확실하지 않았다 의미 :

model.predict(image)[0] 

을 그래서, 한 번 위의 문 을 인쇄를했다 [0]와[0] 한 번하지 않고, 다음과 같이

print model.predict(image) 

[[ 0.99335432 0.00664574]] 

print model.predict(image)[0] 

[ 0.99335432 0.00664574] 

예, 나는 여전히 무슨 뜻인지 확실하지, 외부 대괄호는 후자의 경우에 제거 된 것을 볼 수 있습니다. 어떤 아이디어?

감사합니다.

Answer 1

model.predict return Numpy arrays. API가 다중 입력을 가정하기 때문에 dim = (N, H, W, 3)라고하는 N 개의 이미지가있는 행렬을 제공하면 model.predict은 [[ouptut0], [output1], ... [output (N- 1)]].

입력에 이미지가 하나 밖에 없으므로 입력 크기는 실제로 (1, H, W, 3)이므로 model.predict은 [[ 0.99335432 0.00664574]]입니다.

[0]은 model.predict이 반환하는 첫 번째 요소를 반환하므로 [ 0.99335432 0.00664574]입니다.