keras如何拆分网络层

做深度学习这么久了，好像还没有随笔记过很多东西，随着年龄的增长，没办法，记忆力越来越不行了，只能随笔笔记一下一下了，也顺便把实际的经验分享出去，帮助到更多的人撒，当然了，我也希望有人跟我一起探讨．．．．

今天分享什么呢？算了，一个小问题把，关于深学框架keras网络层的一个应用，好了好了，不废话了．．

我们在做一些图像分割的时候，可能会需要把图像的不同的通道分别处理，这时候就需要把网络层中的某些通道进行拆分，但是keras并没有提供类似`tf.split`这样的方法，反而我们只能看到一个`keras.layers.merge`这样一个合并层，那么我们怎样使用keras进行拆分呢？很简单，借助`keras.layers.core.Lambda`实现

keras在层中间其实都是张量进行传递，那么张量是可以像数组一样进行切片的，知道这一点就好操作了，不好意思，又扯了一大堆废话，直接上代码比较实在

下面以Unet分割网络举例，通过Unet预测图像的mask区域以及图像类别，比如，我们的实际情景是：先预测mask区域，然后再预测mask区域的类别，这时候我们可以通过Unet一步到位，而不用处理2个模型，具体步骤如下:

1.定义数据的label.这里的label不像做图像分类的时候那样，单纯的定义一个0或者1,然后转换为one-hot编码，因为Unet是每个像素级别的分类，所以我们需要为每个像素都定义一个label,换句话说，mask区域有多大，label就有多大,所以第一步是定义一个和mask原图一样大的label数组

2.得到mask区域的label.把mask数组* label数组即可(前提是mask必须是0或1的二值图)

3.把1和2得到的两个数组填充在一个数组里面，返回作为神经网络训练的标签

4. 修改Unet网络结构

代码如下:  

mask_label = np.zeros((self.target_size[0], self.target_size[1], 2))

new_label = np.multiply(mask, mask_label[:, :, 1])   # 求mask区域的label

mask_label[:, :, 0] = mask 　#　第0通道表示mask区域

mask_label[:, :, 1] = new_label  # 第1通道表示mask区域的label                                                                                                                  

Unet网络修改如下:

def slices(x, index): 

      return x[:, :, :, index:index+1]

...

de_out = Conv2D(filters=target_channels, kernel_size=(3, 3), padding='same', strides=(1, 1))(de_8)  # 这层是Unet的倒数第二层，其它层根据自己实际情况构建把

x1 = keras.layers.core.Lambda(slices, arguments={"index": 0}, name="mask")(de_out)  # 拆分网络层得到第0通道

x2 = keras.layers.core.Lambda(slices, arguments={"index": 1})(de_out)  # 拆分网络层得到第1通道

conv10 = keras.layers.multiply([x1, x2], name="mask_label")    # 拆分的2个通道乘积

conv11 = keras.layers.concatenate([x1, conv10])    # 合并通道为一个输出

model = Model(input=[input_layer], output=conv11)

修改完网络层以后，接下来干嘛呢，当然是训练神经网络了．．

所有的步骤就是酱紫啦，好像写的有些啰嗦，代码没有高亮，见谅啊，下次用MarkDown语法改正，我肯定改正哦．．．233333