whether TensorRT supports MobileNet in ONNX framework?

Hi,

I noticed that TensorRT seems not support DepthWise Conv in ONNX framework, so does TensorRT support MobileNet in ONNX framework? if not, when to support MobileNet in ONNX?

Hi,

Only 2D convolution is supported by the onnx parser.

Usually, mobileNet is represented in the TensorFlow format and we do support depthwise conv in uff parser.
If ONNX is essential for you, you can try to add the operation into the onnx parser on your own.
https://github.com/onnx/onnx-tensorrt

Thanks.

Hi,

" If ONNX is essential for you, you can try to add the operation into the onnx parser on your own. "

Sorry, I don’t understand how to add the operation into the ONNX parser, could you explain more?

Hi,

Caffe and ONNX parser is open-sourced:
https://github.com/NVIDIA/TensorRT

So you can just update the ONNX parser to support the depthwise conv layer.
Thanks.