keras支持Lambda可以实现数据自定义处理操作,本文记录切割、截取网络层数据Lambda层实现方法。
需求
需要在网络层中选取某层,截取其中部分特征,用于模型的特征处理。
切割函数
代码语言:javascript复制def cut_map(x,index):
return x[:, index:-index, index:-index, :]
加入网络层
代码语言:javascript复制x = Lambda(cut_map, arguments={'index':32}, name = 'classify_Lambda')(x)
网络效果
使用说明
- 建议该层主要用于训练好的模型的特征处理,直接放入准备训练的模型会造成该层无法传播梯度。