学习记录——StyleGAN2+SA-UNet

SA-UNet for Retinal Vessel improvment using StyleGAN2

  作者提出了一种改进视网膜图像分割的方法，通过创建图像及其相应的分割地图来实现。作者的解决方案包括使用DRIVE数据集1对StylGAN2进行训练，并使用目前在分割DRIVE图像方面取得最先进结果的SA-UNet模型对新合成的图像进行分割。作者使用新生成的图像及其分割结果来重新训练SA-UNet，以获得更好的结果。
  IterNet网络利用U-Net，通过将多个小U-Net连接起来，取得了优势。IterNet的工作方式是首先生成粗糙的分割地图，然后通过在小区域上应用小型U-Net来改进分割地图。

  SA-UNet的主要优势在于它结合了U-Net、SD-UNet和空间注意力块。SA-UNet的架构与U-Net非常相似，但SA-UNet的参数更少，因此训练过程更短，过拟合问题更不容易发生。
  在SA-UNet中，Dropout被DropBlock取代，DropBlock会丢弃感兴趣区域，而不是随机丢弃权重。此外，在任何一对卷积层之间，都有Batch Normalization，如图2所示。此外，SA-UNet还在编码器和解码器之间引入了空间注意力块。

  数据增强一直是计算机视觉领域的一个重要问题。有许多方法可以进行数据增强，从更简单的方法如图像的传统变换（如翻转、裁剪、旋转等）开始。这些方法在一开始就取得了不错的性能，