How to get class label for nvidia OCDNET

Hi,

I am retraining OCDNET model using tao 5.2 for multiple classes. The training is successful. But when I test the model I don’t see the class names in the overlay.

My question is , can we train OCDNET as a multiclass object detection and can we get the object labels when inferenceing.

Regards,
Avijit