yolo v2 using tensorRT on TX2

Can I run yolo v2 on tensorRT? I can successfully convert the yolo v2 weights to caffe. However, I see some of the layers not supported in tensorRT (reorg and region layer params). Can you suggest best way to approach this? Does converting yolo v2 weights to tensorflow and then using tensorRT work?


First, you can find the detail supported layer here:

For non-supported layer, please use Caffe model to get the plugin layer support.
You can implement your custom layer with TensorRT plugin API.

Here is a sample for your reference: