YoloV3 without Batched NMS post processing code

I am using TAO Toolkit version 5.3.0 and has trained a YoloV3 model, which on conversion adds the BatchedNMS layer to it, which was not supported on my embedded device.

So I removed the Batched NMS layer as mentioned in this discussion.

Requesting you help me with the post processing/inference code for the same withou BatchedNMS in it.

Thanks & Regards,
Abhimanyu K S

2 posts - 2 participants

Read full topic