Keras 在网络中自定义切割数据层

2022-08-05 10:37:16 浏览数 (1)

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)

网络效果

使用说明

  • 建议该层主要用于训练好的模型的特征处理,直接放入准备训练的模型会造成该层无法传播梯度。

0 人点赞