通俗易懂理解FCN全卷积网络模型

温故而知新，可以为师矣！

一、参考资料

深度学习笔记（二十三）Semantic Segmentation(FCN/U-Net/PSPNet/SegNet/U-Net++/ICNet/DFANet/Fast-SCNN)

二、FCN相关介绍

1. FCN简介

全卷积神经网络(Fully Convolutional Networks, 简称FCN)，是利用深度学习进行语义分割的开山之作。FCN将传统的分类网络改造成分割网络：替换全连接层为卷积层，利用转置卷积进行上采样获得高分辨率的语义特征图，输出 feature map 的每一维通道预测一个类别的分割结果。

FCN主要用来做 pixel-wise 的 image segmentation 预测，先用传统的CNN结构得到 feature map， 再将传统的 full connected 转换成了对应参数的卷积层。比如，传统pool5层的尺寸是7×7×512，fc6的尺寸是4096， full connected weight 是7×7×512×4096；将全连接层转成卷积层，kernel size 为7×7，input channel 为512，output channel 为4096，这就实现了将传统的网络（分别带有卷积层和全连接层）转成了全卷积网络(fully convolutional network, FCN)。简单理解，FCN全卷积网络就是全部都由卷积层构成的网络，没有FC层。

FCN的一个好处是输入图片尺寸可以任意，不受传统网络全连接层尺寸限制，传统的方法还要用类似SPP结构来避免这个问题。FCN中为了得到 pixel-wise 的prediction，也要把 feature map 通过转置卷积转化到像素空间。

2. FCN网络结构

FCN大致网络结构如下：

上图模型结构为针对VOC数据集的21个语义分割，即数据集包含21种不同分割类型。当图像进入神经网络，第一个卷积层将图像由三通道转换为96通道 feature map，第二个卷积层转换为256个通道，第三个卷积层384个通道，直到最后一个卷积层变为21个通道，每个通道对应不同分割类型。实际上，卷积层整个网络结构中卷积层的通道数可以根据不同任务进行调整，前面每经过一层会对图像进行一次宽高减半的下采样，经过5个卷积层以后，feature map 的尺寸为输入的1/32，最后通过转置卷积将 feature map 宽高恢复到原始输入图像尺寸。

3. FCN网络结构分解

FCN模型结构可以根据分割细粒度使用FCN32s、FCN16s、FCN8s等结构，32s即从32倍下采样的 feature map 恢复到原始输入尺寸，16s和8s则是从16倍和8倍下采样恢复到原始输入尺寸，当然还可以使用4s、2s结构，数字越小使用的转置卷积层进行上采样越多，对应模型结构更加复杂，理论上分割的效果更精细。

3.1 FCN32s模型结构

FCN32s模型结构示意图：

3.2 FCN16s模型结构

FCN16s模型结构示意图：

3.3 FCN8s模型结构

FCN8s模型结构示意图：

4. FCN的特征融合(Add)

在FCN论文中，作者采用对 low-level 和 high-level 特征进行融合的方式。这是因为：

1. 网络比较深的时候，特征图尺寸通常比较小，对这种特征图进行上采样，有很好的语义信息，但分辨率很差。
2. 网络比较浅的时候，特征图尺寸通常比较大（接近input image），对这种特征图进行上采样，有很好的细节，但语义信息很差。

因此，对 low-level 和 high-level 特征进行融合，可以在得到很好的细节基础上，也能获得尽可能强的图像语义信息。

三、相关经验

(MindSpore)代码实现

图像语义分割网络FCN(32s、16s、8s)原理及MindSpore实现

(TensorFlow)代码实现

github代码：tensorflow-fcn

四、参考文献

Long J, Shelhamer E, Darrell T. Fully convolutional networks for semantic segmentation[C]//Proceedings of the IEEE conference on computer vision and pattern recognition. 2015: 3431-3440.