mmdetection
mmdetection copied to clipboard
A question about comment in `mask2former_head`
Hi,
Maybe the shape
here should be (batch_size, num_queries, c)
?
https://github.com/open-mmlab/mmdetection/blob/56e42e72cdf516bebb676e586f408b98f854d84c/mmdet/models/dense_heads/mask2former_head.py#L331
Kindly ping @chhluo have a look
@pixeli99 yes, you are right. I will update it later.