1. 程式人生 > >利用DSB2017冠軍開源代碼為LUNA16生成mask

利用DSB2017冠軍開源代碼為LUNA16生成mask

round targe src ESS 開源 9.png style sub img

代碼地址: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,而且效果應該會更好。

利用DSB2017冠軍開源代碼為LUNA16生成mask