Keras 任意の層の出力を取りたい!!

はじめに

Kerasでネットワーク中の任意の層の出力を得る方法を記述します.

実装例

predict.py

from keras import backend as K
get_1st_layer_output = K.function([model.layers[0].input],
                                  [model.layers[5].output])

layer_output = get_1st_layer_output([img,])
img_layer_5 = layer_output[0][0,:]

実装例 (解説)

K.finction()の引数に入力層と得たい層を指定し. 関数を作成します.
その関数に入力(ここでは画像)を引数として与えると得たい層の
出力が得られます!!