前言
卷积神经网络在医学图像分割上有着广泛的应用,针对视网膜眼底图像的分割成为研究的热门领域。
模型介绍
ensemble learning
12个模型取平均
网络结构如下图:
“Ensemble of Deep Convolutional Neural Networks for Learning to Detect Retinal Vessels in Fundus Images”
平均准确率: 0.9470
组会PPT展示:基于卷积神经网络的光学眼底图像分割研究.pdf
基于U-net的方法
-
原始的U-net方法:
利用对称的编码器——解码器结构
解码部分采用反卷积的方式
padding采用valid无零填充模式,即多余的剔除
特征层个数分别为:64-128-256-512-1024
参考链接:U-Net: Convolutional Networks for Biomedical Image Segmentation
-
改进的方法1:
解码部分采用反卷积方式Deconv2d(),padding=same有零填充形式
特征层:64-128-256-512-1024
“Automatic Brain Tumor Detection and Segmentation Using U-Net Based Fully Convolutional Networks”
github代码实现参考链接:U-Net Brain Tumor Segmentation
-
改进的方法2:
解码过程采用上采样操作UpSampling2dLayer(),扩大分辨率
特征层个数:32-64-256-512
多通路特征融合的方法
-
利用多尺度特征融合进行视网膜分割
“Development of Automatic Retinal Vessel Segmentation Method in Fundus Images via Convolutional Neural Networks”
多尺度输入融合的方法
-
利用多尺度输入进行视网膜目标检测
“Joint Optic Disc and Cup Segmentation Based onMulti-label Deep Network and Polar Transformation”
结合CRF后处理的方法
-
结合条件随机场后处理的方法进行血管分割
“DeepVessel: Retinal Vessel Segmentation via Deep Learning and Conditional Random Field”
我提出的方法
-
基于Unet的改进:多尺度卷积核+多尺度特征输入
示例图如下:
-
阅读并对程序做了一些修改
-
将训练和测试数据转化为Hdf5格式:prepare_data_drive.py
-
图像的预处理:灰度化、归一化、自适应直方图均衡化(CLAHE)、gamma亮度调整
-
训练数据,随机选定像素中心分割成小块(32*32),做为训练输入数据
测试数据,为了能够恢复结果,按顺序依次分割成小块,每块分割好后再拼接
-
训练和测试过程代码还在修改中,是否添加CRF做为后处理?能够对优化函数进行改进?
-