上传者: 42132354
|
上传时间: 2021-11-18 09:37:47
|
文件大小: 13.13MB
|
文件类型: -
unet
unet主要用于语义分割, 这里是一个细胞边缘检测的例子, 数据集比较简单。
unet的网络结构, 因像字母‘U’而得名。
这里有一篇关于unet的 [论文](U-Net: Convolutional Networks for Biomedical Image Segmentation), 论文里面的网络结构如下:
说一下这个网络:
输入572×572×1, 输出:388×388×2, 大小不一样。
主要是因为卷积的过程中, 每次卷积会减小, 在copy and crop中, 也会减小。
我这里设计的网络, 并没有像上图的网络一样, 原封不动的实现出来, 而是借助vgg网络结构来实现的。
看上图, 我们发现, unet的前半部分采用2层卷积+一层池化的设计方式, 这一点和vgg16的前半部分很相似, 因此, 我在实现的过程中, 采用了vgg16的前10层。
网络设计
def vg