关于unet网络医学分割的网址 unet,大家可以在该网站中学习有关unet的知识
我将我的版本上传上了github,这是用keras实现的,运行data.py就可以将图片转换成.npy格式,这里我已经生成数据了,直接运行unet.py就可以了
大家可以在这里下载:https://github.com/huangshaoyin/U-net
(具体参考的是哪个开源项目,我已经忘了,如有抄袭,可以联系我)
打开data.py文件
只要把这些路径改成你电脑对应的位置就行:
现在已经改成了相对路径:
还有unet.py
这些路径也要修改,把路径都修改完了,就可以打开PyCharm运行unet.py
训练图片:
参考图片:
测试图片:
测试结果:
测试结果很差,分析原因是因为训练样本只有30张,这是远远不够的,那篇论文说有做数据增强,以此来增加训练样本,所以接下来可以尝试一下增强数据后的训练结果。