
摘要
医学图像分割是医学图像分析中的一个重要步骤。随着卷积神经网络在图像处理领域的快速发展,深度学习已被应用于医学图像分割,例如视盘分割、血管检测、肺部分割、细胞分割等。此前,基于U-Net的方法已被提出。然而,连续的池化和步幅卷积操作会导致部分空间信息的丢失。本文中,我们提出了一种上下文编码器网络(以下简称CE-Net),用于捕捉更多高层次信息并保留2D医学图像分割的空间信息。CE-Net主要包含三个组成部分:特征编码模块、上下文提取器和特征解码模块。我们使用预训练的ResNet块作为固定的特征提取器。上下文提取器模块由新提出的密集空洞卷积(Dense Atrous Convolution, DAC)块和残差多核池化(Residual Multi-Kernel Pooling, RMP)块组成。我们将所提出的CE-Net应用于不同的2D医学图像分割任务中。综合实验结果表明,该方法在视盘分割、血管检测、肺部分割、细胞轮廓分割以及视网膜光学相干断层扫描层分割等方面优于原始U-Net方法和其他最先进的方法。
代码仓库
David-zaiwang/Image_segmentation_framework
pytorch
GitHub 中提及
Guzaiwang/CE-Net
官方
pytorch
GitHub 中提及
HzFu/MNet_DeepCDR
tf
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| lung-nodule-segmentation-on-luna | CE-Net | Accuracy: 0.99 |
| medical-image-segmentation-on-isbi-2012-em | CE-Net | VInfo: 0.9878 VRand: 0.9743 |
| retinal-vessel-segmentation-on-drive | CE-Net | AUC: 0.9779 Accuracy: 0.9545 |
| retinal-vessel-segmentation-on-rose-1-dvc | CE-Net | Dice Score: 57.83 |
| retinal-vessel-segmentation-on-rose-1-svc | CE-Net | Dice Score: 75.11 |
| retinal-vessel-segmentation-on-rose-1-svc-dvc | CE-Net | Dice Score: 73.00 |
| retinal-vessel-segmentation-on-rose-2 | CE-Net | Dice Score: 70.66 |