代码地址:https://github.com/lfz/DSB2017
先展示下生成的mask与真实mask
subset9
subset8
subset7
subset6
subset5
subset4
subset3
subset2
subset1
上面的图片是从subset1~subset9中随机选取的切片,左边为生成的mask,右边为ground truth,效果相当不错,不过仍有瑕疵,比如说subset3的mask就明显少了一大块。
在DSB2017中其实利用了两部分数据,一部分是比赛方提供的数据,一部分是LUNA16数据集,LUNA16数据集提供了mask,所以代码中是分开处理的,对于LUNA16利用提供的mask,对于比赛数据,采用阈值化加形态学操作,生成mask,那么这个mask有啥用呢,是用来剔除与肺部无关的区域,便于肺结节检测。
DSB2017中具体生成的mask的文件是preprocessing/step1.py。
作为替代方案,其实也可以采用CNN来进行分割,产生mask,而且效果应该会更好。