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?
Hi,
First, you can find the detail supported layer here:
[url]Developer Guide :: NVIDIA Deep Learning TensorRT Documentation
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:
Thanks.