其他分享
首页 > 其他分享> > RetinaNet基础概念

RetinaNet基础概念

作者:互联网

1.目标检测算法

  一般的步骤包括:输入图片--->得到候选框--->根据候选框提取特征-->对框内的特征进行分类(回归)

2. 视觉检测器

  1.1One-stage

    以YOLO和SSD为代表的单级结构,它们摒弃了提取proposal的过程,只用一级就完成了识别/回归,虽然速度较快但准确率稍差。

  1.2Two-stage

    以Faster RCNN为代表的两级识别方法,这种结构的第一级专注于proposal的提取,第二级则对提取出的proposal进行分类。两级结构准确度较高,但因为第二级需要单独对   每个proposal进行分类/回归,速度比较慢。

  那么,有没有只用单级结构又能提高准确率的方法呢?

 

3. focal loss

  针对one-stage在训练时会被易于分类的背景示例所支配。作者提出了一种新的损失函数,它可以作为处理类不平衡的先前方法的一种更有效的替代方法。

  (1)定义cross entropy :

     

  作者提出了一种新的针对二分类问题的loss。其中,y是0,1标签。P是模型输出的概率(即估计y=1的概率)。

  为了符号方便表示:

                     

  即:

        

  (2)Balanced Cross Entropy

    解决类不平衡的常用方法是引入加权因子a,其中a属于[0,1]

       

  (3)定义focal loss

    虽然a平衡positive/negative的案例,但它没有区分easy/hard examples。通过定义一个可调聚焦参数(1 − pt)γ,定义了focal loss:

           

    在实践中,添加了一个a平衡变量,得到以下形式:

            

    总之,本质上讲,Focal Loss 就是一个解决分类问题中类别不平衡、分类难度差异的一个损失函数。比如:在训练时,前景类和背景类之间的极度不平衡(如:1:1000)

 

4.RetinaNet

  RetinaNet是一个统一的网络,由骨干网和两个特定于任务的子网组成。骨干负责计算整个输入图像上的卷积特征图,并且是一种自卷积网络。第一个子网在主干的输出上执行卷积对象分类;第二个子网执行卷积边界框回归。

  (1)FPN (Feature Pyramid Network)

    FPN通过自上而下的路径和横向连接增强了标准卷积网络,因此网络从单个分辨率输入图像有效地构建了丰富的多尺度特征。详细见:        

      https://www.cnblogs.com/huajing/p/13651175.html

  (2) 分类子网

    分类子网预测每个A anchors和K个对象类在每个空间位置处对象存在的概率。

    设计很简单:

      输入是C通道的多层特征图--->4个3×3的卷积核(通道数为C)--->relu激活函数--->3×3卷积核(通道数为A)--->sigmoid激活函数

      (一般C=256,A=9)

 

  (3)box回归子网

    大体上同分类子网,只是最后是一个线性输出。如上图(d)

 https://www.cnblogs.com/huajing/p/13651175.html

标签:loss,子网,RetinaNet,卷积,分类,基础,---,概念,proposal
来源: https://www.cnblogs.com/huajing/p/13656350.html