本文将介绍U-net模型,以及其tensorflow的实现,保存在Github上
U-net 结构
U-net顾名思义,其结构是一个U型的网络
左侧为一个下采样过程,分4组卷积操作(蓝色箭头)进行。每组卷积操作后进行一次maxpool操作(红色箭头),将图片进一步缩小为原来的 1/2 1 / 2 。通过4组操作将 572×572×1 572 × 572 × 1 大小的输入图片,计算为 32×32×1024 32 × 32 × 1024 大小。
右侧的上采样过程。上采样过程使用的是4组反卷积(浅绿色箭头),TensorFlow使用的函数是tf.nn.conv2d_transpose(),每次上采样将图片扩展为原来的