DETR

2022年11月

Posted by franztao on November 23, 2022

DETR:End to End Object Detection with Transformers

DETR训练过程:

第一步用CNN抽特征。

第二步用Transformer编码器去学全局特征,帮助后边做检测。

第三步,结合learned object query用Transformer解码器生成很多预测框。

第四步,匹配预测框与GT框,在匹配上的框里做目标检测的loss。

DETR推理过程:

第一步用CNN抽特征。

第二步用Transformer编码器去学全局特征,帮助后边做检测。

第三步,结合learned object query用Transformer解码器生成很多预测框。

第四步,置信度大于0.7的作为前景物体保留,其余作为背景。