search

lvzongyao / detr-demo

A simple demo of how to use Facebook's DETR object detector for inference. DETR: End-to-End Object Detection with Transformers.
4 stars 1 forks source link

extract CLASS #1

Open changyunke opened 3 years ago

changyunke commented 3 years ago

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 commented 3 years ago

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