MS-DETR
MS-DETR copied to clipboard
关于推理
大佬好,请问当推理的时候是把use_ms_detr设置为False吗?如果设置为False,那么我注意到decoder layer中,两个attention的顺序又变成了先self_attn后cross_attn?
你好~推理的时候也是设置成 True。目前代码里推理的时候 one-to-many 分支还是 forward 了,但是 one-to-many 分支的预测最终没有用到,如果不想 forward 这个分支可以修改下代码(比如把 ms_detr 设成 False 并把 attention layer 的顺序调整为 cross_attn -> self_attn)