pdf code

End-to-End Object Detection with Transformers

这篇paper给的思路是一个用新工具进行复古的思路。在Yolo与RCNN统一之前,object detection的一个做法是使用RNN序列地输出bounding boxes,当时有一个需要注意的trick就是loss function在计算的时候需要先将Ground truth与预测框计算一个最优匹配,然后对应计算loss. 本文的新颖之处就是使用transformer替代RNN并行decode.

网络结构

image

一些细节:

  1. 匹配的时候使用entropy损失+giou+L1loss, 匹配在CPU上实现,使用的是scipy.optimize.linear_sum_assignment
  2. 损失函数使用giou + L1 Loss.
  3. 训练epoch数比较长,在custom dataset上复现难度挺大