UNet 、3D-UNet 、VNet 区别
医学图像的几个常用模型,简单总结一下。
三个model的代码在我的Github上,可以参考一下:https://github.com/VickyLLY/unet_and_vnet
一、UNet
创新点:
U型结构,下采样,上采样;短接通道(skip-connection)
二、3D-UNet
2D结构的U-Net是基本一样,唯一不同:2D操作换成了3D
好处:三维图像就不需要单独输入每个切片进行训练,而是可以采取图片整张作为输入到模型中
详细解读可以看:3D U-Net论文解析
三、VNet
Vnet是针对3D图像提出来模型。
创新:
1、引入残差,水平向的残差连接采用element-wise;
2、卷积层代替上采样和下采样的池化层。
ment-wise;
2、卷积层代替上采样和下采样的池化层。
详细Vnet和Unet对比:Unet和Vnet