今天将继续分享从网络结构上进行改进提出ETVNet模型来分割脑肿瘤。 为了方便大家学习理解整个分割流程，我将整个流程步骤进行了整理，并给出每个步骤的结果，希望对大家有所帮助。
一、Edge Guidance 模块介绍
在论文《ET-Net A Generic Edge-aTtention Guidance Network for Medical Image Segmentation》中，作者提出在网络中加入了边缘注意力来引导分割结果。作者指出在分割网络中的编码模块中，浅层的编码层学习到的是局部的边缘特征，通过Edge Guidance模块来表达边缘信息，此外作者还使用了Weighted Aggregation模块，这个模块是将每个解码模块的输出先进行通道重标定BraTS18——多模态MR图像脑肿瘤分割挑战赛续4，然后再将不同尺度重标定结果进行相加，最后再与Edge Guidance模块的输出进行集成来提高最后的分割结果。Edge Guidance模块和Weighted Aggregation模块结构如下所示。
这两个模块的结构比较简单，这里就不给出代码了，大家可以自己动手实践一下。
二、脑肿瘤图像分析与预处理
（1）、多模态MR脑肿瘤图像分析。
分析的过程基本上跟上一篇一致，这里就不多言了，直接从数据处理开始。
（2）、准备脑肿瘤分割数据。
首先将4个模态序列的MR原始图像进行合并生成4个通道的三维图像，原始图像大小都是（240x240x155x1），合并后大小是（240x240x155x4）；
其次对Mask图像进行one-hot操作，将原始图像大小都是（240x240x155x1），生成大小是（240x240x155x4）：通道0中非零值区域是背景区域，通道1中非零值是坏疽区域，通道2中非零值是浮肿区域，通道3中非零值是增强肿瘤区域；
最后对图像和Mask进行分块——取Patch操作，生成若干个（128，128，64）大小的图像和Mask，判断并输出非零的Mask和对应的图像。
三、脑肿瘤分割
（1）、搭建ETVNet3d模型，与VNet3d模型不同之处在于增加了EdgeGuidance模块和WeightedAggregation模块，网络输入大小是（128x128x64）。
（2）、loss采用的是多类Focalloss，具体实现可以点击原文链接查看具体代码。
（3）、训练的损失函数和精度如下图所示。
（4）、脑肿瘤分割推理过程：首先将原始flair，T1，T2，T1ce图像一起读取进来并进行z-score标准化操作，然后将四个模态图像合并成4通道三维图像（240x240x155x4），输入到网络中去，网络输入大小是（240x240x48x4），在z方向上分块输入并拼接最终得到（240x240x155）分割结果。
（5）、进行了结果测试，左边是金标准图像，右边是预测结果图像，如下所示。
了大家更好的学习，我把ETVNet网络代码分享到github上：
https://github.com/junqiangchen/VNetFamily
如果大家觉得这个项目还不错，希望大家给个Star并Fork，可以让更多的人学习。 如果碰到任何问题，随时留言，我会尽量去回答的。
本文分享自微信公众号 - 最新医学影像技术（MedicalHealthNews）。
如有侵权，请联系 support@oschina.cn 删除。
本文参与“OSC源创计划”，欢迎正在阅读的你也加入，一起分享。