DesenUnet图像大小变化:从输入到输出的详细解析

本文以输入图像大小为[1, 1, 256, 256]为例,详细解析DesenUnet网络中图像大小的变化过程。其中,1表示batch size,1表示通道数,256表示图像的高度和宽度。

1. Encoder模块:

Encoder模块主要进行特征提取和下采样操作,逐步减小图像的空间尺寸,并增加特征图的通道数。

| 操作 | 输入大小 | 输出大小 ||---|---|---|| 输入图像 | [1, 1, 256, 256] | - || 第一次卷积 | [1, 1, 256, 256] | [1, 64, 256, 256] || 第二次卷积 | [1, 64, 256, 256] | [1, 64, 256, 256] || 下采样 | [1, 64, 256, 256] | [1, 64, 128, 128] || 第三次卷积 | [1, 64, 128, 128] | [1, 128, 128, 128] || 第四次卷积 | [1, 128, 128, 128] | [1, 128, 128, 128] || 下采样 | [1, 128, 128, 128] | [1, 128, 64, 64] || 第五次卷积 | [1, 128, 64, 64] | [1, 256, 64, 64] || 第六次卷积 | [1, 256, 64, 64] | [1, 256, 64, 64] || 下采样 | [1, 256, 64, 64] | [1, 256, 32, 32] || 第七次卷积 | [1, 256, 32, 32] | [1, 512, 32, 32] || 第八次卷积 | [1, 512, 32, 32] | [1, 512, 32, 32] || 下采样 | [1, 512, 32, 32] | [1, 512, 16, 16] |

2. Bottleneck模块:

Bottleneck模块是连接编码器和解码器的桥梁,进一步提取高层语义信息。

| 操作 | 输入大小 | 输出大小 ||---|---|---|| 输入特征图 | [1, 512, 16, 16] | - || 第一次卷积 | [1, 512, 16, 16] | [1, 1024, 16, 16] || 第二次卷积 | [1, 1024, 16, 16] | [1, 1024, 16, 16] || 上采样 | [1, 1024, 16, 16] | [1, 1024, 32, 32] |

3. Decoder模块:

Decoder模块与Encoder模块对称,进行上采样操作,逐步恢复图像的空间尺寸,并将特征图的通道数减少到原始图像的通道数。

| 操作 | 输入大小 | 输出大小 ||---|---|---|| 输入特征图 | [1, 1024, 32, 32] | - || 第一次卷积 | [1, 1024, 32, 32] | [1, 512, 32, 32] || 第二次卷积 | [1, 512, 32, 32] | [1, 512, 32, 32] || 上采样 | [1, 512, 32, 32] | [1, 512, 64, 64] || 第三次卷积 | [1, 512, 64, 64] | [1, 256, 64, 64] || 第四次卷积 | [1, 256, 64, 64] | [1, 256, 64, 64] || 上采样 | [1, 256, 64, 64] | [1, 256, 128, 128] || 第五次卷积 | [1, 256, 128, 128] | [1, 128, 128, 128] || 第六次卷积 | [1, 128, 128, 128] | [1, 128, 128, 128] || 上采样 | [1, 128, 128, 128] | [1, 128, 256, 256] || 第七次卷积 | [1, 128, 256, 256] | [1, 64, 256, 256] || 第八次卷积 | [1, 64, 256, 256] | [1, 64, 256, 256] || 最后一次卷积 | [1, 64, 256, 256] | [1, 1, 256, 256] |

综上所述,经过DesenUnet网络的处理,输入图像的大小经过编码器、瓶颈和解码器模块的处理后,最终输出的图像大小仍然为[1, 1, 256, 256]。

DesenUnet图像大小变化:从输入到输出的详细解析

原文地址: https://www.cveoy.top/t/topic/fRJS 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录