Why drop the last dim for classification

facebookresearch / detr

End-to-End Object Detection with Transformers

Apache License 2.0

13.09k stars 2.37k forks source link

Closed idejie closed 1 year ago

idejie commented 1 year ago

Hi,

As the format of coco, the id 0 is non-sense or means background.

I'm confused about why drop the last dim for classification.