extract CLASS

[changyunke]

Thank you for your contribution. i want to know how i can get the CLASS list in detr_demo.ipynb? whether the CLASS list comes from the *.json file of annotation?

[lvzongyao]

Hi, I am not sure if you were referring to the CLASSES list which has length 91. If so, that's the classes from COCO dataset. You might also want to check the official demo https://colab.research.google.com/github/facebookresearch/detr/blob/colab/notebooks/detr_demo.ipynb