[DETR] End-to-End Object Detection with Transformers (ECCV 2020 oral)笔录

基于Transformer的端到端目标检测方法

若本文对您有所助益，烦请给予点赞支持~

论文链接:https://arxiv.org/pdf/2005.12872.pdf
代码链接:https://github.com/facebookresearch/detr

文章结构概述

• 一、transformer
  • • 1. 网络架构
    • 2. scale dot-product attention 及 muli-head attention
    • 3. Attention(Q,K,V)表达式
    • 4. MultiHead(Q,K,V)表达式
    • 5. Position Embedding

  • 二、DETR

  • • 1. 设计初衷
    • 2. 网络架构
    • 2. HungarianMatcher
    • • ① 计算匹配成本
      • ② 计算匈牙利损失
      • ③ 匹配与损失部分核心代码

  • 三、核心代码部分

  • • 1. detr/models/detr.py
    • 2. detr/d2/detr/detr.py
    • 3.detr/models/backbone.py
    • 4.detr/models/transf