参考:https://www.bilibili.com/video/BV1Sh4y1F7ur/?spm_id_from=333.788&vd_source=156234c72054035c149dcb072202e6be
之前的detr正样本数量少,匹配不平衡。
主要修改两个地方:encoder和decoder。
1.在encoder之后加入RPN,atss辅助头。
2.将以上辅助头的输出传到decoder中,作为query。
辅助头的消融实验:
加入更多的头,产生更多的loss,更好优化模型。