草庐IT

TverskyLoss

全部标签

分割网络损失函数总结!交叉熵,Focal loss,Dice,iou,TverskyLoss!

文章目录前言一、交叉熵loss二、Focalloss一、Dice损失函数一、IOU损失一、TverskyLoss总结前言在实际训练分割网络任务过程中,损失函数的选择尤为重要。对于语义分割而言,极有可能存在着正负样本不均衡,或者说类别不平衡的问题,因此选择一个合适的损失函数对于模型收敛以及准确预测有着至关重要的作用。一、交叉熵lossM为类别数;yic为示性函数,指出该元素属于哪个类别;pic为预测概率,观测样本属于类别c的预测概率,预测概率需要事先估计计算;缺点:交叉熵Loss可以用在大多数语义分割场景中,但它有一个明显的缺点,那就是对于只用分割前景和背景的时候,当前景像素的数量远远小于背景像