Keras - model.predict (Bild) [0]

Question

ich über die Aussage kam unten und war nicht sicher, was [0] bedeutet:

model.predict(image)[0] 

Also, ich einen Druck für die obige Aussage einmal tat heraus ohne[0] und einmal mit [0] wie folgt:

print model.predict(image) 

[[ 0.99335432 0.00664574]] 

print model.predict(image)[0] 

[ 0.99335432 0.00664574] 

Ja, ich kann sehen, dass die äußeren eckigen Klammern haben in diesem Fall, aber immer noch nicht sicher, was es bedeutet, entfernt worden. Irgendwelche Ideen?

Danke.

Answer 1

model.predict Numpy-Arrays zurückgeben. Da die API mehrere Eingaben voraussetzt, wenn Sie eine Matrix mit N Bildern angeben, sagen wir dim = (N, H, W, 3), gibt model.predict [[output0], [output1], ... [output (N-)] zurück. 1)]].

In Ihrem Fall gab es nur ein Bild in den Eingängen, so dass die Eingabe Dimension tatsächlich war (1, H, W, 3), daher model.predict zurückgegeben [[ 0.99335432 0.00664574]].

[0] liefert das erste Element von was auch immer model.predict zurückgibt, daher [ 0.99335432 0.00664574].