萌新的卷积神经网络学习笔记--pytorch搭建3D Unet

参考图：

参考图来自论文：https://arxiv.org/pdf/1606.06650.pdf

与2d 的Unet相比两边各少了一层，还有每层卷积通道的变化顺序不同，差不多就是基于2d的Unet，把2d卷积变为3d卷积，2d池化变为3d池化。模型包含了三次(2,2,2)的池化，所以输入图片的大小d,h,w分别都应该为8的倍数。