目标检测:损失函数之SmoothL1Loss

fasterRCNN中的RPN的回归框的loss计算方法

再次深入的解读SmoothL1Loss损失函数:

fasterRCNN中SmoothL1Loss的详解

 

  • 6
    点赞
  • 41
    收藏
    觉得还不错? 一键收藏
  • 8
    评论
多分类目标检测损失函数通常包括类别损失和位置损失两部分。 1. 类别损失函数 常用的类别损失函数是交叉熵损失函数(Cross Entropy Loss)。对于每个目标检测框,我们需要预测它属于哪个类别。假设有 $C$ 个类别,对于第 $i$ 个目标检测框,我们用 $p_{i,j}$ 表示它属于第 $j$ 个类别的概率,用 $y_{i,j}$ 表示它实际上属于第 $j$ 个类别的标签(如果属于第 $j$ 个类别,则 $y_{i,j}=1$,否则 $y_{i,j}=0$)。则交叉熵损失函数可以表示为: $$ L_{cls}=-\frac{1}{N}\sum_{i=1}^{N}\sum_{j=1}^{C}y_{i,j}\log(p_{i,j}) $$ 其中 $N$ 表示目标检测框的数量。 2. 位置损失函数 位置损失函数通常使用 Smooth L1 Loss。对于每个目标检测框,我们需要预测它的位置信息,包括中心坐标 $(x,y)$、宽度 $w$ 和高度 $h$。假设预测的位置信息为 $(x_i,y_i,w_i,h_i)$,实际位置信息为 $(x_t,y_t,w_t,h_t)$,则 Smooth L1 Loss 可以表示为: $$ L_{loc}=\frac{1}{N}\sum_{i=1}^{N}\sum_{j\in\{x,y,w,h\}}smooth_{L1}(i,j) $$ 其中 $N$ 表示目标检测框的数量,$j$ 表示位置信息的维度,$smooth_{L1}(i,j)$ 表示 Smooth L1 Loss,可以表示为: $$ smooth_{L1}(i,j)=\begin{cases} 0.5x_i^2 & if\ |x_i|<1 \\ |x_i|-0.5 & otherwise \end{cases} $$ 综上所述,多分类目标检测损失函数包括类别损失和位置损失两部分,其中类别损失函数通常使用交叉熵损失函数,位置损失函数通常使用 Smooth L1 Loss。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值