kaggle挑战赛——超声乳腺癌分割

kaggle挑战赛——超声乳腺癌分割

今天将分享乳腺癌超声分割实现版本，为了方便大家学习理解整个流程，将整个流程步骤进行了整理，并给出详细的步骤结果。感兴趣的朋友赶紧动手试一试吧。

一、背景

乳腺癌是全世界女性最常见的死亡原因之一。早期发现有助于减少早期死亡人数。

二、超声图像分析与预处理

（1）、780张乳腺超声数据及标注数据可以-获取，其中只有647张乳腺超声数据有分割标注。

（2）、首先将数据分割成训练集合测试集，按照常规分割比例标准，80%用来训练，20%用来测试，这样就有520例数据用来训练，127例用来测试。

（3）、由于520例训练数据还是不足，需要做数据增强操作，对训练数据进行2倍数据增强操作（平移，旋转，翻转等操作），一共得到1560例数据。

（4）、图像与mask进行归一化操作，直接除以255，归一化到0-1范围。

三、分割网络

（1）、搭建VNet2d模型，网络输入大小是（512，512）。

具体实现可以参考​​Tensorflow入门教程（十九）——基于VNet的前列腺分割案例​​。

（2）、loss采用的是二分类的dice函数。

代码实现如下：

def __get_cost(self, cost_name, Y_gt, Y_pred): H, W, C = Y_gt.get_shape().as_list()[1:] if cost_name == "dice coefficient": smooth = 1e-5 pred_flat = tf.reshape(Y_pred, [-1, H * W * C]) true_flat = tf.reshape(Y_gt, [-1, H * W * C]) intersection = 2 * tf.reduce_sum(pred_flat * true_flat, axis=1) + smooth denominator = tf.reduce_sum(pred_flat, axis=1) + tf.reduce_sum(true_flat, axis=1) + smooth loss = -tf.reduce_mean(intersection / denominator) return loss

具体实现可以参考​​Tensorflow入门教程（三十四）——常用两类图像分割损失函数​​。

（3）、分割损失结果和精度经过如下图所示。

（4）、测试图像结果对比

下图是部分测试图像分割结果与金标准结果对比效果图，中间是金标准，右边是预测结果。在测试数据上可以得到平均IOU分数为0.63左右，平均dice分数为0.74。

​

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。