分割任务中常见的损失计算方式以及预测值(pred)和目标值(target)的形状

产生分割任务中常见的损失计算方式以及预测值(pred)和目标值(target)的形状可以总结如下:

常见损失

1. 二分类损失(Binary Cross-Entropy Loss):

  • 任务类型:适用于单类别分割任务,将目标与背景进行二分类。
  • 损失计算方式:使用二分类损失函数(如二进制交叉熵损失)计算预测掩码与目标掩码之间的差异。
  • pred的大小:[batch_size, channel, height, width],其中channel通常为1,表示预测的概率或分数。
  • target的大小:[batch_size, 1, height, width],其中像素值为0或1,表示目标掩码的二分类标签。

2. 多分类损失(Multi-Class Cross-Entropy Loss):

  • 任务类型:适用于多类别分割任务,其中图像可以包含多个不同的目标类别。
  • 损失计算方式:使用多分类损失函数(如交叉熵损失)计算预测掩码与目标掩码之间的差异。
  • pred的大小:[batch_size, num_classes, height, width],其中num_classes表示目标类别的数量,每个通道表示一个类别的概率或分数。
  • target的大小:[batch_size, 1, height, width],其中像素值为类别索引,表示目标掩码的类别标签。

3. 像素级别的损失(Pixel-Level Loss):

  • 任务类型:适用于需要对每个像素进行细粒度分割的任务。
  • 损失计算方式:使用像素级别的损失函数(如平均绝对误差、均方误差等)计算预测像素值与目标像素值之间的差异。
  • pred的大小:[batch_size, channel, height, width],通常为浮点数张量,表示每个像素的预测值。
  • target的大小:[batch_size, channel, height, width],通常为浮点数张量,表示每个像素的目标值。

4. IoU损失(Intersection over Union Loss):

  • 任务类型:适用于需要衡量预测掩码与目标掩码之间重叠程度的任务。
  • 损失计算方式:使用IoU(交并比)损失函数计算预测掩码和目标掩码之间的交集与并集之比。
  • pred的大小:[batch_size, channel, height, width],通常为二值掩码,表示预测的类别或前景。
  • target的大小:[batch_size, channel, height, width],通常为二值掩码,表示目标的类别或前景。

具体计算公式

1. 二分类损失(Binary Cross-Entropy Loss):

  • 公式: loss = − 1 N ∑ i = 1 N ( y i log ⁡ ( p i ) + ( 1 − y i ) log ⁡ ( 1 − p i ) ) \text{loss} = -\frac{1}{N}\sum_{i=1}^{N}(y_i\log(p_i) + (1 - y_i)\log(1 - p_i)) loss=N1i=1N(yilog(pi)+(1yi)log(1pi))
  • 其中, N N N是样本数量, y i y_i yi是目标掩码的二分类标签(0或1), p i p_i pi是预测掩码的概率或分数。

2. 多分类损失(Multi-Class Cross-Entropy Loss):

  • 公式: loss = − 1 N ∑ i = 1 N ∑ c = 1 C y i c log ⁡ ( p i c ) \text{loss} = -\frac{1}{N}\sum_{i=1}^{N}\sum_{c=1}^{C}y_{ic}\log(p_{ic}) loss=N1i=1Nc=1Cyiclog(pic)
  • 其中, N N N是样本数量, C C C是目标类别的数量, y i c y_{ic} yic是目标掩码的类别标签(0或1), p i c p_{ic} pic是预测掩码的类别概率或分数。

3. 像素级别的损失(Pixel-Level Loss):

  • 公式
    • 平均绝对误差(MAE)损失: loss = 1 N ∑ i = 1 N ∣ y i − p i ∣ \text{loss} = \frac{1}{N}\sum_{i=1}^{N}|y_i - p_i| loss=N1i=1Nyipi
    • 均方误差(MSE)损失: loss = 1 N ∑ i = 1 N ( y i − p i ) 2 \text{loss} = \frac{1}{N}\sum_{i=1}^{N}(y_i - p_i)^2 loss=N1i=1N(yipi)2

4. IoU损失(Intersection over Union Loss)

  • 公式: loss = 1 − ∑ i = 1 N ( y i ∩ p i ) ∑ i = 1 N ( y i ∪ p i ) \text{loss} = 1 - \frac{\sum_{i=1}^{N}(y_i \cap p_i)}{\sum_{i=1}^{N}(y_i \cup p_i)} loss=1i=1N(yipi)i=1N(yipi)
  • 其中, N N N是样本数量, y i y_i yi是目标掩码的二值掩码, p i p_i pi是预测掩码的二值掩码, ∩ \cap 表示交集, ∪ \cup 表示并集
  • 26
    点赞
  • 22
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值