U-Net

U-Net的基本结构

image-20221225175200290

如图所示,U-Net是由卷积神经网络、下采样模块、上采样模块以及短路连接构成。

U-Net的左侧被称为编码器,右侧被称为解码器。

左边做的卷积都是卷积核大小为3×3的卷积,没有padding,卷积步长stride=1,因此我们可以看到图像的尺寸长和宽每经过一次卷积都会缩小2。

右边的上采样采用了反卷积的方法。短路连接会把左边的高分辨率特征图剪切成相等的大小拼接到右边。最后输出的全连接网络是用1×1的卷积实现的。