ashkamath / mdetr

Apache License 2.0
969 stars 125 forks source link

Generate tokens_positive for just COCO for pre-training!! #104

Closed h-pal closed 8 months ago

h-pal commented 8 months ago

I'm trying to pre-train the Mdetr architecture on just the COCO dataset, however, there is no availability of tokens_positive in the COCO annotations. I was wondering if there's a way to generate tokens_positive for the COCO dataset. Any help is appreciated!! Thanks.