上一篇说到如何分割出肾脏区域,接下来就要对肾肿瘤区域进行分割提取。按照以往的经验是需要结合2d和3d网络来进行分割处理的。
1、Vnet2d肾肿瘤分割
肾肿瘤分割初步尝试:采用Vnet2d来实现的,图像输入大小是512x512,在1080ti显卡上跑可以设置batchsize为6,如果大家出现OOM错误,请修改该参数,如果显存比11G还大的话,也可以设置大一些。
2、测试数据上分割结果
把0-209例肾肿瘤数据全部用来训练,选择80%来训练,20%来验证测试。在测试的时候,直接输入原始图像大小512x512。然后计算20%验证数据的dice值,并计算加权平均,结果是0.872。结果看上去并不是那么糟糕。
3、代码及训练模型分享
修改后的代码已经更新到github上了,相应训练好的模型我也分享到百度云盘上:
https://pan.baidu.com/s/1zh23ptvx-tPHeAA9Cp-Png
提取码:r4dj