当前位置:   article > 正文

语义分割——U-net、U-net+++、U-2net学习笔记(1)_细胞语义分割u-net复现代码

细胞语义分割u-net复现代码

目录

一、语义分割:逐像素的二分类问题

二、U-net网络

一、语义分割:逐像素的二分类问题

1、损失函数:采用逐像素的交叉熵损失函数(Cross Entropy Loss)

为了更高效的使样本的预测值逼近真实值,可考虑在损失函数中加入样本均衡、样本数量等因素。交叉熵损失函数公式如下:

Loss=-[ylog\hat{y}+(1-y)log(1-\hat{y}))]

①考虑正样本和负样本的比例,针对样本的前景和背景比例不一样,使得样本中每个像素点的重要程度不一样,在原本的交叉熵损失函数加入权重项 posweight,可以加强对重要的像素点的学习,加入权重项的交叉熵公式如下:

posweight=\frac{numneg}{numpos}

Loss=-[posweiht\ast y_{true}\ast logy_{pred}-(1-y_{true})\ast log(1-y_{pred})]

②针对要学习的样本困难程度不同,可设置参数r,具体的交叉熵公式如下:

Loss=-[(1-y_{pred})^{r}\ast y_{true}\ast logy_{pred}-y_{pred}^{r}(1-y_{true})\ast log(1-y_{pred})]

 ③结合样本数量a的权值的交叉熵公式如下:

Loss=-[a\ast (1-y_{pred})^{r}\ast y_{true}\ast logy_{pred}-(1-a)y_{pred}^{r}(1-y_{true})\ast log(1-y_{pred})]

一般交叉熵的值越小,模型预测效果就越好。

2、评价指标:IOU、MIOU

IOU(intersection over Union,交并比):对某一类别预测结果和真实值的交集与并集的比值。

MIOU(Mean Intersection over Union,平均交并比):对每一类预测的结果和真实值的交集与并集的比值,求和再平均的结果,一般当作多任务分割的评价指标。

二、U-net网络

1、作用:主要解决小目标问题的图像分割

2、网络结构:

 

特点:相比于其他语义分割网络,网络结构简单,因此适合做一些小目标的分割。

结构:

①引入了特征拼接操作,使得原始的浅层特征和后面的复杂特征相互融合,减少特征的丢失;

②网络包括上采样和下采样两个部分,也就是编码和解码的过程。

③学习的U-net代码来源:https://github.com/mateuszbuda/brain-segmentation-pytorch

(此为深度学习小白的学习笔记,可能目前学习不够深入,存在一些问题,虚心求教)

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小小林熬夜学编程/article/detail/87770
推荐阅读
相关标签
  

闽ICP备14008679号