arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2306.02901eess.IVcs.CV

一种用于非配对多模态视网膜图像合成的基于血管分割的 CycleGAN

A Vessel-Segmentation-Based CycleGAN for Unpaired Multi-modal Retinal Image Synthesis

FAU Erlangen-Nürnberg(埃尔朗根-纽伦堡大学)

Aline Sindel, Andreas Maier, Vincent Christlein

AI总结:

本文扩展 CycleGAN,在 UNet vision transformer 生成器前加入血管分割网络,并改造 identity loss、引入分割损失,实现可保留血管结构的非配对多模态视网膜图像合成。

摘要:

视网膜图像的非配对图像到图像转换可以有效扩充基于深度学习的多模态视网膜配准方法的训练数据集。我们的方法通过扩展 CycleGAN 框架,将血管分割网络集成到图像到图像转换任务中。该分割网络被插入到 UNet vision transformer 生成器网络之前,并作为两个域之间的共享表示。我们重新表述了原始的 identity loss,以学习血管分割与真实图像之间的直接映射。此外,我们加入了一个分割损失项,以确保伪图像与真实图像之间的血管位置共享。实验中,我们的方法呈现出视觉上逼真的外观,并保留了血管结构,这是为图像配准生成多模态训练数据的先决条件。

英文摘要:

Unpaired image-to-image translation of retinal images can efficiently increase the training dataset for deep-learning-based multi-modal retinal registration methods. Our method integrates a vessel segmentation network into the image-to-image translation task by extending the CycleGAN framework. The segmentation network is inserted prior to a UNet vision transformer generator network and serves as a shared representation between both domains. We reformulate the original identity loss to learn the direct mapping between the vessel segmentation and the real image. Additionally, we add a segmentation loss term to ensure shared vessel locations between fake and real images. In the experiments, our method shows a visually realistic look and preserves the vessel structures, which is a prerequisite for generating multi-modal training data for image registration.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑