search

mlpc-ucsd / PRTR

(CVPR 2021) PRTR: Pose Recognition with Cascade Transformers
Apache License 2.0
141 stars 29 forks source link

How do you discard unmatched queries #12

Closed ziqi123 closed 2 years ago

ziqi123 commented 2 years ago

Hello, In training stage of person detection Transformer, how do you discard unmatched queries ? I don't see the code of this stage. Is the "person per image" in your code equals to 5 or less in End-to-end variant?

zx1239856 commented 2 years ago

During training, the unmatched queries are not discarded. Different loss functions are applied for matched and unmatched queries. Please read the main paper for details.

For the end-to-end variant, the person per image depends on your GPU memory. If working with a smaller memory size, one way is to randomly sample some person boxes.