肺结节提取和分类【2】
前面的博文讲了肺结节遮罩的提取,对于肺结节提取和分类整个项目来说,主要有两个核心网络:
- 肺结节提取网络,用于从肺实质中提取出可疑肺结节,这个网络需要的数据集是肺实质图像(为了降低其他CT图像的影响)、以及对应的肺结节遮罩。使用的网络主要是Unet和一些改进过的Unet
- 肺结节良恶性分类网络:这里可以用简单的二分类网络,一些常见的例如vgg、res等都可以,但其实可以考虑将HU值这个参数融合进去。
这两个网络训练达到一定的精度之后,就是GUI的设计和实现,我主要用了pyqt5和designer做了一个简单的界面。
肺结节提取网络效果如下:
GUI界面大致如下