李沐论文精读系列:DETR(End-to-End Object Detection with Transformers) Transformer的端到端的目标检测
我们提出了一种将目标检测视为直接集合预测问题的新方法,我们的方法简化了监测管道,有效地消除了对许多手工设计组件的需求,例如非极大值抑制或者锚生成,这些组件有效地编码了我们对任务的先验知识。新框架的主要成分,称为检测转换器或 DETR,是基于集合的全局损失,它通过二部匹配强制唯一的预测,以及变压器编码器-解码器架构。给定一组固定的学习对象查询,DETR 推理对象和全局图像上下文的关系,以并行直接输出最终的预测集。与许多其他现代检测器不同,新模型在概念上很简单,不需要专门的库。
原创
2024-01-09 21:42:04 ·
496 阅读 ·
0 评论