UNet 、3D-UNet 、VNet 区别

UNet 、3D-UNet 、VNet 区别

医学图像的几个常用模型,简单总结一下。

三个model的代码在我的Github上,可以参考一下:https://github.com/VickyLLY/unet_and_vnet

一、UNet

UNet 、3D-UNet 、VNet 区别

创新点:

U型结构,下采样,上采样;短接通道(skip-connection)

二、3D-UNet

UNet 、3D-UNet 、VNet 区别

2D结构的U-Net是基本一样,唯一不同:2D操作换成了3D

好处:三维图像就不需要单独输入每个切片进行训练,而是可以采取图片整张作为输入到模型中

详细解读可以看:3D U-Net论文解析

三、VNet

UNet 、3D-UNet 、VNet 区别

Vnet是针对3D图像提出来模型。

创新:

1、引入残差,水平向的残差连接采用element-wise;

2、卷积层代替上采样和下采样的池化层。

ment-wise;

2、卷积层代替上采样和下采样的池化层。

详细Vnet和Unet对比:Unet和Vnet