search

HDETR / H-PETR-3D

[CVPR2023] This is an official implementation of paper "DETRs with Hybrid Matching".
Other
27 stars 3 forks source link

about infer quert num #2

Open shaoqb opened 1 year ago

shaoqb commented 1 year ago

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

shaoqb commented 1 year ago

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

PkuRainBow commented 1 year ago

@LittlePey Please help to answer this question.

LittlePey commented 1 year ago

@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.