大花蕙兰

首页 » 常识 » 问答 » KiTS19肾肿瘤分割挑战赛四
TUhjnbcbe - 2021/5/13 14:30:00
北京看白癜风哪间医院效果好 http://m.39.net/pf/bdfyy/

之前由于某些原因没有实际运行github上面代码,确实也存在一些Bug,这几天花了点时间调试运行,终于在五一前将肾器官分割结果跑通了,接下来将分享相应结果。

1、Vnet肾器官分割训练效果

我是在ti显卡上跑的,所以batchsize设置成3,如果大家出现OOM错误,请修改该参数,如果显存比11G还大的话,也可以设置大一些。如下图所示是训练loss结果及精度结果。

2、在测试数据上分割结果

前面也说过,训练是用0-例数据来训练,测试是在-上来测试。在测试的时候,我们直接输入原始图像大小,例如xx48,将训练好的模型导入后,将xxn(可能是几百)的输入按z方向取48,不断输入到模型中得到最后肾器官分割结果。如下图所示是部分分割结果对比示意图。

在10个测试集上计算dice值,平均值是95.98%。

3、代码及训练模型分享

修改后的代码已经更新到github上了,相应训练好的模型我也分享到百度云盘上:

1
查看完整版本: KiTS19肾肿瘤分割挑战赛四