bs, l, c, h, w = srcs.shape

megvii-research / AnchorDETR

An official implementation of the Anchor DETR.

Other

338 stars 36 forks source link

bs, l, c, h, w = srcs.shape #44

Closed feimadada closed 2 years ago

feimadada commented 2 years ago

hi, the transformer module, bs, l, c, h, w = srcs.shape, what does the second dimension "l" mean?

tangjiuqi097 commented 2 years ago

Hi, it means the number of features. We only use a single DC5 feature so the "l" is 1. We keep this dimension to facilitate scaling to multiple features if it is needed.