【文件属性】:
文件名称:unet_keras:unet_keras使用图像语义分割
文件大小:13.13MB
文件格式:ZIP
更新时间:2021-04-30 22:37:58
keras unet Python
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
【文件预览】:
unet_keras-master
----src()
--------mask.png(6KB)
--------test.png(167KB)
--------unet.png(78KB)
----unet.py(6KB)
----LICENSE(1KB)
----README.md(4KB)
----data()
--------train()
--------test()
----.gitignore(11B)