目标检测 DETR（2020）

news2024/11/19 10:45:37

文章目录

前言
backbone
位置编码（二维）
encoder、decoder
prediction heads
损失函数计算

前言

DETR全称是Detection Transformer，是首个基于Transformer的端到端目标检测网络，最大的特点就是不需要预定义的先验anchor，也不需要NMS的后处理策略（少了这两部分可以少很多的超参数和计算），用集合的思想回归出100个query之后再用匈牙利算法二分图匹配的方式得到最终的正样本和负样本，第一个实现了端到端的目标检测。

DETR的总体框架如下，可以分为三个部分，分别是：用于抽取图像的特征表示的backbone、使用了transformer的编码器解码器以及prediction heads：FFN，前馈神经网络给出最后的类别和Box信息的预测。

首先通过backbone提取图像的特征，随后将特征信息加上位置编码送入encoder，这里的encoder进行特征序列构造。在decoder中会首先初始化100个向量（object queries)，这些向量通不断学习encoder构造的特征序列，再经过prediction heads得到类别和边界框的预测，最后并通过二分匹配计算损失来优化网络。
在这里插入图片描述

backbone

主干特征提取网络采用resnet50，假设输入为[batch_size, 3, 800, 1066]，经过resn et50，此时输出为[batch_size, 2048, 25, 34]，输出通道为变为2048，图像高和宽都变为原图像高和宽的 1/32，再经过 1 × 1 卷积后进行通道降维，得到的输出为[batch_size, 256, 25, 34]，最后经过patch embedding将每个像素（每个patch块）映射为向量形式，得到[850，batch_size, 256]作为encoder的输入之一。

位置编码（二维）

其实就是原Transform的position embedding的思想，经过backbone得到的特征图的shape为[850，batch_size, 256]，为每一个像素点的对应的feature上的每个位置都添加上位置信息。

正余弦位置编码：
在这里插入图片描述

pos代表不同的patch图片一维向量，i代表一维向量的第i个位置，d_model代表一维向量的总长度（即通道数），采用二维位置编码，通道数的前128维表示x的坐标信息，后128维表示y的坐标信息。
在这里插入图片描述
这样得到每个patch的位置编码Position Encoding，shape与特征图的一样为[850，batch_size, 256] ，与input embedding相加得到encoder的输入。

encoder、decoder

编码器结构基本与transformer相同，input embedding+positional encoding操作转换的图像序列作为transformer encoder的输入，经过encoder得到提取出来的特征图像序列。

在decoder上，解码器使用到一个非常重要的可学习模块，即下图呈现的object queries，使用一个可学习的object query对加强后的有效特征层进行查询，Decoder的输入分为三部分：Encoder的输出Object queries（query_embed）、queries（tgt）。tgt是一个大小为100*256、初始全为0的参数，仅第一层Decoder输入为0，之后几层Decoder的输入是上一层Decoder的输出；object queries是query_embed，是模型的一个可学习编码信息参数，由网络迭代学习到的一个参数），是用nn.Embedding初始化了100个可学习的( x , y ) ，维度是[2,100] ，然后经过位置编码将位置信息搞成256维（实际是[256,100]），其中前128+128维分别表示x，y的坐标信息；100表示模型最多预测出100个目标框，256与图像特征通道数一致可保证注意力机制的正常运算。Decoder对应Object queries的输出经Prediction heads后将用于计算损失、预测框坐标、预测类别。

object queries是预定义的目标查询的个数，代码中默认为100。它的意义是：根据Encoder编码的特征，Decoder将100个查询转化成100个目标，即最终预测这100个目标。最终预测得到的shape应该为[N, 100, C]，N为Batch Num，100个目标，C为预测的100个目标的类别数+1（背景类）以及bbox位置（4个值），再经过prediction heads得到类别和边界框的预测。
在这里插入图片描述

prediction heads

prediction heads是DETR的分类器与回归器，其实就是对decoder提取出来的100个目标进行FFN操作，FFN采用全连接，经过FFN分别得到种类和边界框参数，将100个预测结构object predictions和真实框ground truth box之间通过匈牙利算法进行二分匹配，一个真实框只匹配一个预测结果，其它的预测结果作为背景进行拟合。最后通过反向传播来优化模型参数。
在这里插入图片描述
匈牙利算法