Video inferencing shows category IDs instead of class names

[Kamal-Moha]

I did object detection using detectron2 on a video. The output video does the detection, but it indicates category ids of either 1 or 2. But I want my detection to show the class names of either 'weed' or 'crop' instead.

Instructions To Reproduce the Issue:

1. Here's the code to do the video inferencing:

%run detectron2/demo/demo.py --config-file /content/config2.yaml --video-input /content/crop-weed.mp4 --confidence-threshold 0.6 --output output_colab.mp4 \
--opts MODEL.WEIGHTS /content/output/model_final.pth

Expected behavior:

I expected detectron2 to do the video inference and show the class names of either 'weed' or 'crop' around the bounding box.

Environment:

• Google Colab

[Programmer-RD-AI]

Is this an issue with the visualization of the predictions or the predictions themselves?