主要是将 queries 的 content 部分(以下代码中的 tgt)初始化为 non-object class,并且加入值为 0 的 indicator 向量用作指示这部分 queries 是做匈牙利匹配任务的。 def prepare_for_dn(dn_args, embedweight, batch_size, training, num_queries, num_classes, hidden_dim, label_enc): """ prepare for d...
首先是main.py中进行模型构建,DN-DETR一个主要思想便是引入了噪声,将真实数据加入噪声后送入Decoder来辅助匈牙利算法完成匹配过程,加快训练收敛。故而,这是一个通用的方法,因此,DN-DETR代码中分别对DAB-DETR,Deformable-DETR等都加入了DN处理,在证明其通用性的同时还检验其效果。 构建模型代码在models/DN-DAB-DETR...