Kerasのラムダの書き方の覚え書き
- ラムダ式の中にはバッチ次元まで含めて書く
- output_shapeにはバッチ数を除いたshapeを書く
- 例) imgの最初の4チャンネルのみ抽出したい場合
from keras.layers import Lambda y = Lambda(lambda x: x[:, :, :, :4], output_shape=(shape[0], shape[1],4))(img)
from keras.layers import Lambda y = Lambda(lambda x: x[:, :, :, :4], output_shape=(shape[0], shape[1],4))(img)