AutoAssign

label分配主要是为了计算Loss,这篇文章使得gt内部的所有点同时具有前景和背景类标签和置信度,因为作者认为gt内部的所有点并不应该都认为是正样本,因为有很多背景点

结构跟FCOS是一样的:
在这里插入图片描述

(1)center weighting先验分支

一个分支,根据gt的位置,得到gt框在不同level上的feature(仅仅为了存放权重),并根据高斯分布生成权重,作为中心先验

(2)positive weight map

对于gt框内部的点,根据每个点的分类、回归loss,生成权重map
1.根据回归loss生成回归置信度
在这里插入图片描述
2.根据分类loss生成分类置信度
在这里插入图片描述
就是ImpObj分支*分类分支
3.正样本置信度=分类置信度*回归置信度
在这里插入图片描述
4.再加个指数函数,让置信度变化更剧烈
在这里插入图片描述
5.与center weighting先验相乘
在这里插入图片描述

6.每个点的正样本权重=这个点的置信度/gt框内所有点的置信度的和
在这里插入图片描述
注意:Sn是所有Level上的所有点,所以最后权重会指示到level级别,有的Level上权重高,别的level上权重就低,这就是作者说的scale和Spatial维度的自动Assign

(3)Negative weight map

1.对于gt外的点,负样本权重设置为1
2.对于gt内的点,负样本权重为该anchor和所有gt的max iou的函数
在这里插入图片描述
在这里插入图片描述
并且f(iou)被归一化到[0,1]
在这里插入图片描述在这里插入图片描述

Loss计算

在这里插入图片描述
gt框内所有点的正样本权重*正样本置信度+所有点的负样本权重*负样本置信度(这里有个1-)
正样本置信度包含分类、回归Loss,负样本置信度只包含分类Loss。
分类Loss用GIoU Loss,回归Loss用Focal Loss。

最终实际可视化效果却没有达到作者最开始的设想:
在这里插入图片描述

实际:
在这里插入图片描述
可以看到spatial和scale的自动Assign是实现了,但是这个圆却并不太好看,看起来用混合高斯比这种单高斯会好,这就是IQDet的内容了

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值