深度学习的分割方法

-小透明-

已于 2023-06-12 14:48:28 修改

阅读量1.8k

点赞数 1

分类专栏：计算机视觉文章标签：深度学习人工智能

于 2023-06-05 16:32:36 首次发布

本文链接：https://blog.csdn.net/qq_54809548/article/details/130992115

版权

计算机视觉专栏收录该内容

23 篇文章 0 订阅

订阅专栏

FCN：基于深度学习的语义分割模型

Fully Convolutional Networks，全卷积网络

其核心思想为：

使用不包含全连接的全卷积网络，所以可以输入任意大小的图片。
使用跳跃连接，以结合不同尺度的图像特征，使得模型具备更高的精度与鲁棒性。
使用反卷积对提取后的低分辨率图像进行上采样，输出与原图同分辨率的分割图片，保留了原始图像的空间信息，从而能够对原图的每一个像素都进行预测。

(20条消息) 图像分割（二）：模型（FCN）_fcn模型_米之炼金术师的博客-CSDN博客

语义分割的定义：对像素进行精细化的分类。

用深度学习来解决语义分割，所面临的主要问题是：

早期的深度模型用于分类，输出一维向量，无法分割
深度模型不够精细

动机

如何让网络的可以用于分割？

让网络输出二维特征就可以了

如何令早期神经网络输出二维图像？

去掉全连接层。

如何让模型的输出足够精细？

不精细的原因是？

经过多层卷积池化，feature map的分辨率较低。

比如224*224的图像输出为7*7的feature map，很显然无法很精细

一种可行的方法就是，再把7*7给变大

具体方法就是反卷积。

模型结构

由于1/32分辨率太低，直接分割很粗糙，因此先扩大再与1/16的拼在一起...(yolov3后面的思路灵感来源于此）

FCN的pytorch实现

前面就是一个常规的CNN网络，将一张图像变成feature map。而后，进行反卷积，得到原图尺寸。

层	输出尺寸
输入图像	224×224
卷积1	224×224
池化1	112×112
卷积2	112×112
池化2	56×56
卷积3	56×56
池化3	28×28
卷积4	28×28
池化4	14×14
卷积5	14×14
反卷积6	224×224
输出	224×224

实现细节

反卷积

结果展示

反池化和反卷积：只是把尺寸恢复了，卷积（或池化）过程中丢失的信息并没有恢复

下采样（encoding）下采样（decoding）

UNet

DeepLab系列语义分割

FCN存在的问题

池化层的存在导致细节信息丢失
空间不变性（在左边、右边不对分类结果有影响）对分割任务不友好

难点1: 细节信息丢失

本质上，语义分割是一种包含了低维语义特征的任务。

对于边缘、纹理、颜色等信息较为敏感。

而池化层的存在导致了这些细节信息丢失，即便进行上采样也无法恢复。

之前讨论过，池化的意义在于对信息进行浓缩，从而扩大感受野，提高信息的维度。

那么如何才能在不损失细节信息的情况下，尽量扩大感受野？

解决方案1: 空洞卷积

空洞卷积可以在不进行池化的前提下，尽可能地扩大感受野，从而快速提高信息的浓缩程度。

练习：计算上述模型的最终感受野

空洞卷积的优势：扩大感受野，保存细节信息

空洞卷积的劣势：小物体不够健壮

难点2: 空间不变性

CNN的一个重大优势在于空间不变性。

同一个物体，在图像的不同位置，不同形态，不同角度，都应当输出相同的值。

例如，一张猫的图像，总会输出猫类别的one hot向量。

但是对于分割来说，这种空间不变形会带来不便性。

对应解决方案：CRF

引入了fully connected CRF，从原图作为输入，结合特征图进行了优化。

CRF

条件随机场 (conditional random field)

前置知识

CRF是一种判别模型，利用条件概率图模型建模条件概率�(�|�)完成判别任务。

也就是说，CRF是对一个条件概率进行估计。

CRF用于语义分割

动机

图像分割任务的核心目的在于对每个像素都分配一个标签。

然而，由于池化卷积的处理方式，导致目标区域的边缘存在较为模糊的地方。

那么就需要CRF为边缘提供新的信息，从而得到更好的边缘。

建立随机场

给定一张图像：

* 定义 $X=\{X_1, X_2, ..., X_N\}$, 其中$X_i$为第i个像素的预测标签；
* 定义 $L=\{L_1, L_2, ..., L_N\}$, 其中$L_i$为第i个像素的真实标签；
* 定义 $I=\{I_1, I_2, ..., I_N\}$, 其中$I_i$为第i个像素数据。