萌新的卷積神經(jīng)網(wǎng)絡(luò)學(xué)習(xí)筆記--pytorch搭建3D Unet

參考圖:

參考圖來(lái)自論文:https://arxiv.org/pdf/1606.06650.pdf
與2d 的Unet相比兩邊各少了一層,還有每層卷積通道的變化順序不同,差不多就是基于2d的Unet,把2d卷積變?yōu)?d卷積,2d池化變?yōu)?d池化。模型包含了三次(2,2,2)的池化,所以輸入圖片的大小d,h,w分別都應(yīng)該為8的倍數(shù)。
標(biāo)簽:卷積神經(jīng)網(wǎng)絡(luò)Unet