weixin_57920621的博客

原创 手搓Unet部分的一些尝试

在写Unet网络的部分之前需要知道VGG16网络的架构，因为Unet借鉴了VGG16前面部分的结构。然后根据Unet的网络结构图，大致可以写得代码如下。为了方便，卷积块和反卷积块的结构都写得比较简单。Unet借鉴了除了全连接层外，即最后一个最大池化层之前的所有结构，以下为Unet网络结构。