about infer quert num

[shaoqb]

The paper says that the one-to-one branch is used for inference. Where is it reflected in the code?

[shaoqb]

Or you only use all_cls_scores and all_bbox_preds in your inference, but the number of queries is 2700?

[PkuRainBow]

@LittlePey Please help to answer this question.

[LittlePey]

@shaoqb Hi, to only use one-to-one branch in inference, you can set num_query=900, comment out Line494~Line498 and set self_attn_mask to None in Line499. To load ckpt trained with one-to-many, the num_query in Line328 should be 2700 and you need take the first 900 query in reference_points before using reference_points.