解决目标检测中密集遮挡问题——Repulsion loss

解决目标检测中密集遮挡问题——Repulsion loss


之前参加了df的钢筋检测比赛,比赛中的一个难点是密集遮挡问题,为了解决这个问题,参考了之前旷视针对人群检测中密集遮挡问题而提出的Reploss,这篇论文收录于cvpr2018。我把Reploss的思路拿过来用在钢筋检测中,效果还是不错的,在这里记录一下这篇论文。
在目标检测中,遮挡问题是比较常见的,主要分为两种,一种是待检测目标之间相互遮挡,一种是待检测目标与被干扰物遮挡,
第一种遮挡又分为类间遮挡和类内遮挡,类内遮挡产生于同类物体,也被称为密集遮挡。第二种遮挡,由于目标干扰物体遮挡,而算法只能学习待检测物体的特征,因此第二种遮挡只能通过增加样本来优化检测效果。
密集遮挡的问题:由于密集遮挡的两个目标的类别是相同的,所以两个目标之间的特征是相似的,检测器很可能无法定位。本应该属于目标A的proposal很可能会想目标B发生偏移,导致定位不准确,而目标B本身有自己的proposal,在接下来的nms中,目标A的proposal很可能被目标B的proposal所抑制,进而造成了目标A的漏检。因此,对于密集遮挡问题,NMS的阈值时很敏感的,阈值过高,造成误检,阈值低,造成漏检。
目前的目标检测器都是利用回归来进行定位,位置回归的目标是让proposal和gtbox尽可能的接近,而不考虑周围的物体。如果预测框周围有其他物体时,对其并没有任何的惩罚。

Rep loss

Reploss收到磁石吸引排斥的启发,Reploss包含三个目标:包括一个吸引项和两个排斥项,吸引项的要求是预测框靠近其目标&

  • 9
    点赞
  • 72
    收藏
    觉得还不错? 一键收藏
  • 8
    评论
遮挡损失函数(repulsion loss)是一种用于目标检测任务的损失函数,主要用于解决目标之间的遮挡关系。该损失函数的代码实现如下: ```python import torch def repulsion_loss(pred_boxes, target_boxes, repulsion_threshold): ''' 遮挡损失函数(repulsion loss)的计算方法 参数: - pred_boxes: 预测的边界框,形状为[N, 4],N表示边界框的个数,每个边界框包含了(xmin, ymin, xmax, ymax)四个坐标 - target_boxes: 真实的边界框,形状为[N, 4] - repulsion_threshold: 遮挡阈值,表示当两个边界框之间的IOU大于该值时,认为存在遮挡关系 返回值: - loss: 遮挡损失 ''' num_boxes = pred_boxes.size(0) loss = 0.0 # 计算边界框之间的IOU矩阵 iou = get_iou(pred_boxes, target_boxes) # 遍历所有预测边界框 for i in range(num_boxes): # 找到与当前预测框IOU大于阈值的真实框 mask = iou[i] > repulsion_threshold num_repulsions = torch.sum(mask) if num_repulsions == 0: # 如果不存在遮挡,损失值为0 continue # 计算遮挡损失 repulsion_loss = torch.sum(iou[i][mask]) loss += repulsion_loss / num_repulsions return loss ``` 上述代码首先定义了一个名为`repulsion_loss`的函数,该函数接受预测框、真实框和遮挡阈值作为输入,并返回计算得到的遮挡损失。 函数首先获取预测框的数量,并定义一个变量`loss`用于保存遮挡损失的累加值。 接下来,利用辅助函数`get_iou`计算预测框和真实框之间的IOU(交并比)矩阵。 然后,遍历所有预测框,对于每个框,找到与之IOU大于阈值的真实框,即存在遮挡关系的真实框。如果不存在遮挡,损失值为0,继续遍历下一个框。 如果存在遮挡关系的真实框,计算遮挡损失,即将这些真实框与当前预测框之间的IOU值累加得到。最后,将遮挡损失除以遮挡框的数量,得到一个平均的遮挡损失值。 最后,返回计算得到的遮挡损失。 这就是遮挡损失函数的代码实现。
评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值