Sparse R-CNN: End-to-End Object Detection with Learnable Proposals (CVPR, 2021)

Sparse R-CNN: End-to-End Object Detection with Learnable Proposals

paper:https://arxiv.org/pdf/2011.12450.pdf
code:https://github.com/PeizeSun/SparseR-CNN

文章结构概览

• 1. 摘要

  • 2. 研究背景

  • 3. 稀疏R-CNN

  • • 3.1 整体流程
    • 3.2 功能模块
    • 3.3 集合预测的匹配与损失函数
    • 3.4 阅读代码后的再思考

  • 4. 实验分析

  • 5. 简洁代码实现

  • • 引言基础 RoIPooler 融合FPN的代码解析
    • 5.1 SparseR-CNN/projects/SparseRCNN/sparsercnn/detector.py
    • 5.2 SparseR-CNN/projects/SparseRCNN/sparsercnn/loss.py
    • 5.3 SparseR-CNN/projects/SparseRCNN/sparsercnn/head.py
    • • 5.3.1 动态头模块
      • 5.3.2 R-CNN头模块
      • 5.3.3 动态卷积层