在之前的文章中我分享了Tensorflow的基本知识内容,接下来我将会分享如何利用Tensorflow将深度学习应用到医学图像上,今天我会分享深度学习在医学图像分割的实际案例。
1、Unet模型
在医学图像处理领域,有一种应用很广泛的全卷积神经网络模型结构——Unet,网络模型结构如下。关于该网络具体的细节,请阅读相关资料:https://lmb.informatik.uni-freiburg.de/people/ronneber/u-net/,这里我就不多说了。
2、问题分析
我采用的是网上公开的细胞图像病理数据,数据地址:https://warwick.ac.uk/fac/sci/dcs/research/tia/glascontest/download/。
3、网络实现
我是参考了网上Tensorflow版本的Unet来实现分割模型,地址:https://github.com/jakeret/tf_unet。
4、效果
由于数据集很小只有165组图像,所以数据增强是必须的图像预处理步骤。下面是模型训练时损失函数变化的情况。
在测试图像上,我用训练好的模型来进行分割,整个测试集上的平均精度为90.75%。为了更主观的查看模型的效果,我给出了2组图像(原始细胞图像和原始细胞分割图像)并用模型分割后的对比效果图。
测试的原始细胞图像A
测试的原始细胞分割图像A
模型的分割图像A
测试的原始细胞图像B
测试的原始细胞分割图像B
模型的分割图像B
从上面可以看到整体分割效果还是不错的,感兴趣的就自己动手试一试吧。如果碰到任何问题,随时留言,我会尽量去回答的。