xingyizhou / CenterNet2

Two-stage CenterNet
Apache License 2.0
1.19k stars 189 forks source link

Inference using demo.py issue #75

Open shoxa-mir opened 2 years ago

shoxa-mir commented 2 years ago

Hello, When I try to inference using demo.py file I am getting all the results however, I get only class_ids on top of the bounding boxes. How can I modify the code to put class_name instead?

I have registered dataset using detectron2 tutorial and using coco format json annotations for my custom dataset.